C - Good Subarrays
题意: 给个长度为 n 的数组,选取一个区间,使得其区间和等于区间的长度,问有多少这种区间。
思路: 暴力直接O(n2) 的去找肯定超时,所以我们做个转化,将所有元素减一,那么所有区间和等于0的区间都是满足条件的区间。所以我们可以在维护前缀和时,记录下每个前缀和出现的次数,当两个前缀和相等,那么其之间的区间和必为0。然后累加计数即可。
Code:
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e6+7;
int a[N];
int main()
{
int T;cin>>T;
while(T--){
int n;
cin>>n;
string s;
cin>>s;
map<int,int> mp;
for(int i=0;i<n;i++) a[i+1] = s[i]-'0'-1;
LL sum=0,ans=0;
mp[0] = 1;
for(int i=1;i<=n;i++){
sum+=a[i];
if(mp[sum]) ans += mp[sum];
mp[sum]++;
}
cout<<ans<<endl;
}
return 0;
}