Educational Codeforces Round 93 (Rated for Div. 2) C. Good Subarrays

最新推荐文章于 2025-11-16 16:39:56 发布

原创最新推荐文章于 2025-11-16 16:39:56 发布 · 239 阅读

1 ·

CC 4.0 BY-SA版权

思维专栏收录该内容

154 篇文章

订阅专栏

本文介绍了一种解决区间和为0问题的有效算法。通过将原问题转化为数组元素减一后的区间和问题，利用前缀和的概念，我们能够快速找到所有符合条件的子数组。文章提供了详细的代码实现，包括初始化、输入处理、核心算法逻辑和输出结果。

题目链接
思路：
将每一位上的数组-1，将问题转化为求区间和为0的问题，再维护一个前缀和即可。
代码：

#include<bits/stdc++.h>
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const int N=210;
const int M=2e4+5;
const double eps=1e-8;
const int mod=998244353;
const int inf=0x7fffffff;
const double pi=3.1415926;
using namespace std;
signed main()
{
    IOS;
    int t;
	cin>>t;
	while (t--)
	{
		int n;
		cin>>n;
		string s;
		map<int,int>mp;
		cin>>s;
		int sum=0,ans=0;
		mp.clear();
		for (int i=0;i<n;i++)
		{
			sum+=s[i]-'0'-1;
			if(sum==0)
                ans++;
			ans+=mp[sum];
			mp[sum]++;
		}
		cout<<ans<<endl;
	}
	return 0;
}