设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;
}
}