Codeforces 1398C-C Good Subarrays 思维前缀和

题目链接：https://codeforces.com/contest/1398/problem/C；
题目大意：给出一个数组，求数组中有几个区间，满足区间和为 区间长度。

首先，求区间和，就肯定要用到前缀和。
(前缀和参考
https://blog.csdn.net/qq_41661809/article/details/86727017)
然而直接用前缀和暴力遍历，时间复杂度达到了O(n²)，会TLE。

#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
int x[100000];
int main(){
	int t;
	x[0] = 0;
	cin>>t;
	while(t--){
		int n, sum = 0;
		char a;
		cin>>n;		
		for(int i = 1; i <= n; i++){
			cin>>a;
			x[i] = a - '0' + x[i - 1];
		}
		for(int i = 1; i <= n + 1; i++){
			for(int j = 0; j < i; j++){
				if(x[i] - x[j] == i - j) sum++;
			}
		}
		cout<<sum<<endl;
		a = getchar();
	}
	return 0;
}

而实际上，区间和等于长度可以看作区间的平均值为一。因此，若将数组每个元素减一，题目就变为求区间和为零的区间。用map记录前缀数组每个元素的值出现的次数。若前缀和之前出现，则中间就有区间和为零的区间，个数为该区间和之前出现的次数。时间复杂度为O(n)。以下为ac代码。

#include<iostream>
#include<algorithm>
#include<map>
using namespace std;
typedef long long ll;
map<int, int> m;
int x[1000000];
int main(){
	int t;
	x[0] = 0;
	cin>>t;
	while(t--){
		ll n, sum = 0;
		m.clear();
		char a;
		cin>>n;		
		for(int i = 1; i <= n; i++){
			cin>>a;
			x[i] = a - '0' + x[i - 1] - 1;
			m[x[i]]++;
			if(x[i] == 0) sum += m[x[i]];
			else sum += m[x[i]] - 1;
		}
		cout<<sum<<endl;
		a = getchar();
	}
	return 0;
}

顺便一提，虽然n只有1e5，但出现次数却会爆int，真是坑人…