codeforces educational 93 C

题目

设si为到i点的前缀和（即s[1]加到s[i]），题目要求我们求满足s[i]-s[j]=i-j的个数，那我们可以对于每一个i，求有多少个满足条件的j，答案就出来了。（对于力扣的区间和为k的个数那题，也可以用这种方法）。

我们把式子变形为s[i]-i=s[j]-j，就很好做了。要注意一点，我们要在0的位置先+1，否则对于前缀就是答案的case，是无法统计进去的（力扣那题也一样）

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<vector>
#include<cmath>
#include<map> 
#include<string>
#include<queue>
#include<stack> 
#include<bitset>
#include<list>
#include<set>
#include<utility>
#include<iomanip>
#define IO ios::sync_with_stdio(false)
#define eps 1e-7
#define int long long
using namespace std;
int a[100005],sum[100005],n,t,ans;
char s[100005];
signed main()
{
	cin>>t;
	while(t--)
	{
		map<int,int>mp;
		ans=0;
		cin>>n;
		cin>>s+1;
		int len=strlen(s+1);
		memset(sum,0,sizeof(sum));
		mp[0]++;//要注意一点，我们要在0的位置先+1，否则对于前缀就是答案的case，是无法统计进去的
		for(int i=1;i<=len;i++)
		{
			sum[i]=sum[i-1]+s[i]-'0';
			ans+=mp[sum[i]-i];
			mp[sum[i]-i]++;
		}
		cout<<ans<<endl;
	}
}