C. Good Subarrays(老套路思维了)

吐槽一句:这种题型遇到过好多次

几乎每一次都是枚举区间开头位置计算结果

---

考虑现在区间是[l,r],考虑加入a[r+1]对这个区间合法性的影响

$区间长度每增加1就需要一个0进来抵消$

$当数字为1时,加不加这个1无所谓,加入这个1,区间长度加1,值也加1$

$当数字为2时,如果加入这个2,区间长度加1,值加2,此时需要有0来填充$

$我们把0看作-1,表示加入0对区间的影响是负面的$

$同理把1看作0,表示加入1对区间是没影响的$

$把数字x看作x-1,表示加入数字x需要区间内还有x-1个0才能抵消$

此时我们要找的合法区间,就是区间和为0的区间了

$剩下的就简单了,先用map记录一遍所有前缀和$

$比如,令当前位置的前缀和是x,就map[x]++$

$接下来枚举每个位置作为开头,由于区间结尾大于开头$

$所以计算答案时先map[x]--,因为这个位置之前的前缀和不能使用(区间末尾>开头)$

$然后答案加上map[x],这个map[x]表示后面的前缀和是x的次数$

$这就是区间和为0的个数$

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
int t,n,sumn[maxn],top;
char a[maxn];
map<int,int>mp;
int main()
{
	cin >> t;
	while( t-- )
	{
		mp.clear();
		long long ans=0;
		scanf("%d%s",&n,a+1);
		int qian=0;
		for(int i=1;i<=n;i++)
		{
			int w=a[i]-'0'-1;
			qian+=w;
			mp[qian]++;
		}
		qian=0;
		ans=mp[0];//前缀和本来就是0的也可以作为答案,可以看作区间开头是0 
		for(int i=1;i<=n;i++)
		{
			int w=a[i]-'0'-1;
			qian+=w;
			mp[qian]--;//先把[1,i]的前缀和消除掉,我们想要的区间结尾要大于i才行 
			ans+=mp[qian];	
		}
		cout << ans << "\n";
	}
}