思维题真___?

前缀和, sum第一次出现某个值时(非0), 说明他是开头, 所以最后mp[sum]++,可以想象成为记录点, 几个点能构成几条线段?
而sum == 0的时候, 额外 ans + 1, 因为对于0来说, 开头一个数没有也是0, 所以sum == 0总是比别人多一个点.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#include <bits/stdc++.h>
using namespace std;
using ll = long long;

int t;
char s[100010];
ll sum, ans;

map<int, int>mp;

int main() {
    cin >> t;
    int n;
    while (t--) {
        cin >> n;
        scanf("%s", s);
        mp.clear();
        ans = sum = 0;
        for (int i = 0; i < n; i++) {
            sum += (s[i] - '0') - 1;
            if (sum == 0) ans += 1;
            ans += mp[sum];
            mp[sum] ++;
        }
        cout << ans << '\n';
    }
    return 0;
}