C - Good Subarrays (Round 93 div 2 思维)

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;
}
©️2020 CSDN 皮肤主题: 我行我“速” 设计师:Amelia_0503 返回首页