C. Good Subarrays 前缀和思维题

题目

   借鉴

• 题意

给你一个数组，让你找 $\left({\sum }_{i=l}^r{a}_i=r-l+1\right)$ 的区间的数量。

• 思路

所以用一个 $map$ 存一下 $sum[i]-i$ ，然后每个贡献就是 $C_x^2$ 。特别要处理一下 $sum[i]-i==0$ 的情况 ，这时候贡献就不是 $C_x^2$ ，因为不等于零时，我们是考虑在这几个里面任意选两个，就相当于是造了一个区间，当 $0$ 的情况，可以 $[0,i]$ 这个区间

• 代码

#include<bits/stdc++.h>

using namespace std;
typedef long long ll;
const int N=2e5+10,mod=1e9+7;
int a[N];
#define PII pair<int,int>
#define x first
#define y second
#define PB push_back
#define eps 1e-6

void solve()
{
	int n;cin>>n;
	string s;cin>>s;
	map<int,int> mp;
	vector<int> vec(n+1);
	for(int i=0;i<n;i++){
		vec[i+1]=(s[i]-'0');
	}

	mp[0]++;//针对0的情况特殊处理，相当于添加位置为0的前缀和为0，选择到了这个点，就相当于【0，x】这个区间。
	for(int i=1;i<=n;i++){
		vec[i]+=vec[i-1];
		mp[vec[i]-i]++;
	}
	ll ans=0;
	for(auto item:mp){
		//cout<<item.x<<" "<<item.y<<endl;
		ans=ans+1ll*(item.y)*(item.y-1)/2;
	}

	cout<<ans<<endl;
}
int main()
{
#ifndef ONLINE_JUDGE
	freopen("in.txt", "r", stdin);
	freopen("out.txt", "w", stdout);
#endif
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	int t;
	cin >> t;
	while (t--)
		solve();
}