【JZOJ A组】括号序列

Description

Input

Output

Sample Input

输入1：
aabaab
输入2：
abcabcabc
输入3：
aabbcc

Sample Output

输出1：
4
输出2：
0
输出3：
6

Data Constraint

思路

暴力做法：维护一个栈，如果栈顶相同，则代表配对，将其弹出栈，只要栈为空，ans++

优化：可以发现，若位置i和位置j，i<j，栈相同，则i–j是合法的括号序列。所以答案就可以直接加上（至于怎么加，详见代码）
开一个map（hash）统计每个栈出现了几次即可

代码

#pragma GCC optimize(3)
#include<iostream>
#include<cstdio>
#include<cstring>
#include<map>
using namespace std;
const int maxn=1e6+7;
map<string,int> a;
char st[maxn],s[maxn];
long long ans;
int main()
{
	freopen("bracket.in","r",stdin); freopen("bracket.out","w",stdout);
	scanf("%s",st);
	int l=strlen(st),t=0;
	for(int i=0; i<l; i++) s[i]='0';
	a[s]=1; 
	for(int i=0; i<l; i++)
	{
		if(!t)
		{
			s[++t]=st[i]; a[s]=a[s]+1; ans+=a[s]-1;
//			printf("i=%d s=%s a[s]=%d cnt=%lld ans=%lld\n",i,s,a[s],cnt,ans);
			continue;
		}
		char c=s[t];
		if(c==st[i]) s[t--]='0';else s[++t]=st[i];
		a[s]=a[s]+1;
		ans+=a[s]-1;
//		printf("i=%d s=%s a[s]=%d cnt=%lld ans=%lld\n",i,s,a[s],cnt,ans);
	}
	printf("%lld",ans);
}