Good Subarrays 思维

题意

若子数组的元素之和等于元素个数，那么这个子数组就为好子数组。那么给你一个子数组序列字符串，判断该序列有多少个好子数组。

思路

长度为len的子段，数字和为len
那么平均每一位上是1
将每个数 -1之后 ，得前缀和
那么 长度为len的子段，数字和为len 可转化为 长度为len的子段，数字和为0

首先 对于前缀和a[pos]已经为0的（l=1 r=pos） 直接加入答案里

其次 若a[pos1]=a[pos2]（pos2>pos1)
那么 a[pos2]-a[pos1]=0 即在区间 l=pos1+1 r=pos2 的数字和为0
所以 对于每个a[pos] 答案要加上 在它之前与它相等的前缀和的个数

代码

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <map>
#define mm 1000000007
using namespace std;
typedef long long ll;
map<int,int>mp;
int a[100010];
char s[100010];
int main()
{
    int t,n;
    scanf("%d",&t);
    while(t--)
    {
        mp.clear();
        scanf("%d",&n);
        scanf("%s",s+1);
        for(int i=1;i<=n;i++)
        {
            a[i]=s[i]-'1';
        }
        a[0]=0;
        ll ans=0;
        mp[0]=1;
        for(int i=1;i<=n;i++)
        {
            a[i]+=a[i-1];
            if(mp.count(a[i])){ans+=mp[a[i]];mp[a[i]]++;}
            else mp[a[i]]=1;
        }
        printf("%lld\n",ans);
    }
    return 0;
}