AtCoder Regular Contest 099 题解

题意：

给出一个操作序列包含$<>+-$，分别是下标左移右移，当前位置加减。问有多少对$(i,j)$满足只做这里面的操作，结果和做完所有操作一样。

题解：

很神奇的一道题。
首先将操作后得到的序列看成一个多项式：$T(S)={\sum }_{i=-10^9}^{10^9}{a}_i{x}^i$
考虑在操作序列前加操作，多项式会变成什么
$$t(>S)=T(S)x$$

$$t(<S)=T(S)x^{-1}$$

$$t(+S)=T(S)+1$$

$$t(-S)=T(S)-1$$

给整个多项式一个hash值c，那么就是问多少对$(i,j)$满足这样得出来的序列的hash等于c。
这四个操作显然是可逆的，设$(i,j)$中的操作为$t_{s_i}{t}_{s_{i+1}}{t}_{s_{i+2}}\dots t_{s_j}$逆操作为$t_{s_i}^{-1}{t}_{s_{i+1}}^{-1}{t}_{s_{i+2}}^{-1}\dots t_{s_j}^{-1}$
$$t_{s_i}{t}_{s_{i+1}}{t}_{s_{i+2}}\dots t_{s_j}(0)=c$$

$$t_{s_n}^{-1}{t}_{s_{n-1}}^{-1}{t}_{s_{n-2}}^{-1}\dots t_{s_i}^{-1}{t}_{s_i}{t}_{s_{i+1}}{t}_{s_{i+2}}\dots t_{s_j}(0)=t_{s_n}^{-1}{t}_{s_{n-1}}^{-1}{t}_{s_{n-2}}^{-1}\dots t_{s_i}^{-1}(c)$$

$$t_{s_n}^{-1}{t}_{s_{n-1}}^{-1}{t}_{s_{n-2}}^{-1}\dots t_{s_{j+1}}^{-1}(0)=t_{s_n}^{-1}{t}_{s_{n-1}}^{-1}{t}_{s_{n-2}}^{-1}\dots t_{s_i}^{-1}(c)$$
变成两个后缀形式，可以线性求出hash值，map统计即可。
自己写的怎么都过不了，各种hash被卡，抄了别人写法
WA：

#include<map>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL long long
#define MP make_pair
using namespace std;
const LL mod=1e9+7;
LL base[4],pre[4][500010],inv[4];
struct node{LL a,b;}op[4][250010];
map<pair<pair<LL,LL>,pair<LL,LL> >,LL> mp;
LL pow(LL a,LL b)
{
	LL ans=1;
	while(b)
	{
		if(b&1) ans=ans*a%mod;
		a=a*a%mod;b>>=1;
	}
	return ans;
}
LL n,p=0,t[500010],c[4],a[4][250010],b[4][250010];
char s[250010];
void Pre()
{
	for(int k=0;k<4;k++)
	{
		LL cnt=1;op[k][n+1].a=1;
		for(LL i=n;i>=1;i--)
		{
			op[k][i]=op[k][i+1];
			if(s[i]=='<') cnt=cnt*base[k]%mod,(op[k][i].a*=base[k])%=mod;
			if(s[i]=='>') cnt=cnt*inv[k]%mod,(op[k][i].a*=inv[k])%=mod;
			if(s[i]=='-') (op[k][i].b+=cnt*pre[k][n]%mod)%=mod;
			if(s[i]=='+') op[k][i].b=((op[k][i].b-cnt*pre[k][n]%mod)%mod+mod)%mod;
		}
	}
}
void solve(LL *a,LL c,LL k) {for(LL i=1;i<=n;i++) a[i]=(c*op[k][i].a%mod+op[k][i].b)%mod;}
int main()
{
	base[0]=2333;base[1]=10037;base[2]=19260817;base[3]=998244353;
	for(int i=0;i<4;i++) inv[i]=pow(base[i],mod-2);
	scanf("%lld",&n);getchar();
	scanf("%s",s+1);
	for(LL k=0;k<4;k++)
	{
		pre[k][0]=1;for(LL i=1;i<=2*n;i++) pre[k][i]=pre[k][i-1]*base[k]%mod;
	}
	for(LL i=1;i<=n;i++)
	{
		if(s[i]=='<') p--;if(s[i]=='>') p++;
		if(s[i]=='-') t[p+n]--;if(s[i]=='+') t[p+n]++;
	}
	for(int k=0;k<4;k++)
		for(LL i=-n;i<=n;i++) (c[k]+=pre[k][i+n]*t[i+n]%mod)%=mod;
	Pre();
	for(int i=0;i<4;i++) solve(a[i],0,i);
	for(int i=0;i<4;i++) solve(b[i],c[i],i);
	LL ans=0;
	for(LL i=2;i<=n+1;i++)
	{
		mp[MP(MP(b[0][i-1],b[1][i-1]),MP(b[2][i-1],b[3][i-1]))]++;
		ans+=mp[MP(MP(a[0][i],a[1][i]),MP(a[2][i],a[3][i]))];
	}
	printf("%lld",ans);
}

AC：

#include<map>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LLu unsigned long long
using namespace std;
char s[250010];
LLu base=233,inv=7204522363551799129,f[250010],X=1,Y,ans=0;
int n;
map<LLu,int> mp;
int main()
{
	scanf("%d",&n);getchar();
	scanf("%s",s+1);
	for(int i=1;i<=n;i++)
	{
		f[i]=f[i-1];
		if(s[i]=='+') f[i]+=X;
		if(s[i]=='-') f[i]-=X;
		if(s[i]=='<') X*=base;
		if(s[i]=='>') X*=inv;
		mp[f[i]]++;
	}
	X=1;
	for(int i=1;i<=n;i++)
	{
		ans+=mp[f[n]*X+Y];
		if(s[i]=='+') Y+=X;
		if(s[i]=='-') Y-=X;
		if(s[i]=='<') X*=base;
		if(s[i]=='>') X*=inv;
		mp[f[i]]--;
	}
	printf("%llu",ans);
}