Good Subarrays(前缀和)

题意：给你一串字符串，问有多少个子串满足str[r]-str[l]=r-l

很明显要用到前缀和，但暴力一个一个找肯定是不行滴，这里要用一个小技巧，将所有的数减去一，那么str[r]-str[l]=r-l就变成了str[r]-str[l]=0,所以满足条件的话两位置之间的和应该为0，所以我们只需要寻找sum[x]==sum[y]的个数就可以算出结果，结果是(sum[x]-1)*sum[x]/2,我们可以便判断边处理，遇到sum[x]==0时ans++，然后再加上这之前的sum[x]的个数。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<map>
#include<iostream>

using namespace std;

const int N = 1e5 + 10;

long long arr[N];
long long sum[N];

map<long long,long long> mp;

string str;

int main() {
	ios::sync_with_stdio(false);
	int t;
	cin >> t;
	while (t--) {
		long long n, ans = 0;
		mp.clear();
		memset(sum, 0, sizeof(sum));
		cin >> n;
		cin >> str;
		for (int i = 1; i <= n; i++) {
			arr[i] = str[i - 1] - '0';
			sum[i] = sum[i - 1] + arr[i]-1;
			if (sum[i] == 0) ans++;
			ans += mp[sum[i]];
			mp[sum[i]]++;
		}
		cout << ans << endl;
	}
}