C. Good Subarrays(思维+前缀和)

Problem - 1398C - Codeforces

题意:

 

给你一个由0到9的整数组成的数组a1,a2,...,an。如果一个子数组al,al+1,al+2,...,ar-1,ar的元素之和等于这个子数组的长度（∑i=lrai=r-l+1），这个子数组就是好的。

例如，如果a=[1,2,0]，那么有3个好子数：a1...1=[1],a2...3=[2,0]和a1...3=[1,2,0]。

计算数组a的好子阵列的数量。

输入
第一行包含一个整数t（1≤t≤1000）--测试案例的数量。

每个测试案例的第一行包含一个整数n（1≤n≤105）--数组a的长度。

每个测试用例的第二行包含一个由n个小数位组成的字符串，其中第i个数字等于ai的值。

保证所有测试案例的n之和不超过105。

输出
对于每个测试案例，打印一个整数--数组a的良好子数组的数量。

例子
input
3
3
120
5
11011
6
600005
输出
3
6
1
注意
语句中考虑了第一个测试案例。

在第二个测试案例中，有6个好的子数：a1...1, a2...2, a1...2, a4...4, a5...5和a4...5。

在第三个测试案例中，只有一个好的子阵列：a2...6。

题解:

问一段的数字和相加等于这一段的长度的子数组有多少个

每一个位置应该都有一个'1',像5 0 0 0 0

我们可以每遍历一位sum+b[i]-1,代表这位数应该对此位做的贡献

如果sum = 0的话,代表从1开始到i每一位上都有一个'1',因为b[1]加到b[i]  - i = 0

如果不等于0,假设为 3,我们看3是否出现过,如果出现过代表什么?

代表中间一段的值相加的结果 = 这一段的长度 (因为减了该长度的1)

#include<iostream>
#include<algorithm>
#include<map>
#include<queue>
using namespace std;
int a[200050];
char b[100050];
void solve()
{
	int n;
	cin >> n;
	cin >>b+1;
	map<int,int> f;
	int s = 0;
	long long ans = 0;
	for(int i = 1;i <= n;i++)
	{
		s += b[i]-'0' - 1;
		if(s == 0)
		ans ++;
		ans += f[s];
		f[s]++;
	} 
	cout<<ans<<"\n";

}
int main()
{
	int t = 1;
	cin >> t;
	while(t--)
	{
		solve();
	}
}
//5 3 4