Codeforces C - Good Subarrays（前缀和 + Map）

题目 链接

题意简析

题目给定了一个序列，然后需要求出一个子序列（连续），他们的和等于他们的长度。
前缀和暴力枚举自然是T的，这里有一个思路，首先将所有的元素自减1，那么a[i] + a[i + 1] + a[i + 2] + … + a[i + n - 1] = n 就转化成 a[i] - 1 + a[i + 1] - 1 + a[i + 2] - 1 + … + a[i + n - 1] - 1 = 0。这就意味着我们只需要统计子串和为0的方案数即可。
我们计算1…n的前缀和，如果当前的sum为零，那么就是一种方案，当然还需要加上前面已经计数过的方案，因为 0 - 0 = 0，下面给出例子。
（原始序列，数值并没有减少）
1 0 0 3
（转化序列，数值减少）
0 -1 -1 2
（前缀和序列）
0 -1 -2 0
那么，第一个【1】就是一种方案，然后最后的【1 0 0 3】是一种方案，同时序列④ - ①也是一种方案，即【0 0 3】。
同理，如果sum不等于零，也是相同的情况。
（原始序列）
1 0 0 2
（转化序列，数值减少）
0 -1 -1 1
（前缀和序列）
0 -1 -2 -1
第一个【1】是一种方案，【0 2】又是一种方案，因为有④ - ②， -1 - (-1) = 0。
所以，我们只需要记录前缀和出现的次数，每次采取右减左就能完成计数。另外，在和为0的时候额外加一种方案即可。

参考代码

#include <bits/stdc++.h>
using namespace std;
const int N = 2E2 + 10;
int a[N];

int main() {
	// freopen("in.txt", "r", stdin);
	int t;
	cin >> t;
	while (t--) {
		int n;
		cin >> n;
		string s;
		cin >> s;
		long long ans = 0;
		int sum = 0;
		map<int, int> m;
		for (int i = 1; i <= n; i++) {
			int x = s[i - 1] - '0' - 1;
			sum += x;
			if (sum == 0) ans++;
			ans += m[sum];
			m[sum]++;
		}
		cout << ans << endl;
	}
	return 0;
}