题目

有一个塔,他的名字叫做粽粑,粽粑的每一层都有一个颜色 .
粽粑非常厉害,它在吸收天地精华之后会长高.粽粑的长高方式有两种:
1.在塔顶长出一层.
2.在塔底长出一层,即原来的第一层变成第二层,第二层变成第三层,以此类推,新长出来的是第一层.
粽粑有可能在某个时刻不是很开心,这个时候它会撤销它的前若干次长高.
你现在想知道粽粑长高的奥秘,于是找到了粽粑,发现它的入口上写着这么一句话:要进入粽粑,请找出一段最长的塔的区间,满足翻转后颜色不变.
粽粑会不断的长高(或撤销),所以它每次长高后你都要回答.为了你的方便,粽粑一开始的高度为0.

分析

我发现我的字符串学傻了⊙﹏⊙‖∣
回文树模板题,但是卡不过。
假设上一次的答案为ans,
对于头尾,我们分别维护长度为ans的字符串的两个哈希值(这个长度为ans的字符串的哈希值高位分别在前或后),
看哈希值是否一样就能判断是否为回文串。每次长高只会让ans+1或+2,判断一下。

#pragma GCC optimize(3)
#include <cmath>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
#include <map>
const int maxlongint=2147483647;
const long long mo=1e9+7;
const int N=10000005;
using namespace std;
int n,m,ans[N],top;
long long sum[N][4];
char t[N];
short int num[N*3];
long long mi[N],ny,val;
long long pow1(long long x,int y)
{
	long long sum=1;
	for(;y;x=x*x%mo,y>>=1)
		if(y&1) sum=sum*x%mo;
	return sum;
}
int main()
{
	scanf("%d\n",&n);
	ny=pow1(101,mo-2);
	int l=N,r=l-1;
	mi[0]=1;
	for(int i=1;i<=n;i++) mi[i]=mi[i-1]*101%mo;
	for(int i=1;i<=n;i++)
	{
		char c,c1,c2;
		c=getchar(),c1=getchar(),c2=getchar();
		int x=((c1-'0')*10+c2-'0'+(ans[top]!=0?ans[top]:0))%100;
		if(c=='1')
		{
			num[++r]=x,t[++top]='1';
			ans[top]=ans[top-1];
			sum[top][0]=sum[top-1][0],sum[top][1]=sum[top-1][1],sum[top][2]=sum[top-1][2],sum[top][3]=sum[top-1][3];
			long long v1=sum[top][0],v2=sum[top][1],len=ans[top];
			if(len) v1=((v1-mi[len-1]*num[r-len]%mo+mo)*101%mo+num[r])%mo;
			if(len) v2=((v2-num[r-len]+mo)%mo*ny%mo+mi[len-1]*num[r])%mo;
			sum[top][0]=v1,sum[top][1]=v2;
			v1=(v1+mi[len]*num[r-len])%mo;
			v2=(v2*101+num[r-len])%mo;
			if(v1==v2)
			{
				ans[top]=len+1;
				sum[top][0]=v1,sum[top][1]=v2;
				sum[top][2]=(sum[top][2]+mi[len]*num[l+len])%mo;
				sum[top][3]=(sum[top][3]*101+num[l+len])%mo;
			}
			if(r-l+1>=len+2)
			{
				v1=(v1+mi[len+1]*num[r-len-1])%mo;
				v2=(v2*101+num[r-len-1])%mo;
				if(v1==v2)
				{
					ans[top]=len+2;
					sum[top][0]=v1,sum[top][1]=v2;
					sum[top][2]=(sum[top][2]+mi[len]*num[l+len])%mo;
					sum[top][3]=(sum[top][3]*101+num[l+len])%mo;
					sum[top][2]=(sum[top][2]+mi[len+1]*num[l+len+1]%mo)%mo;
					sum[top][3]=(sum[top][3]*101+num[l+len+1])%mo;
				}
			}
		}
		else
		if(c=='2')
		{
			num[--l]=x,t[++top]='0';
			ans[top]=ans[top-1];
			sum[top][0]=sum[top-1][0],sum[top][1]=sum[top-1][1],sum[top][2]=sum[top-1][2],sum[top][3]=sum[top-1][3];
			long long v1=sum[top][2],v2=sum[top][3],len=ans[top];
			if(len) v1=((v1-mi[len-1]*num[l+len]%mo+mo)*101%mo+num[l])%mo;
			if(len) v2=((v2-num[l+len]+mo)%mo*ny%mo+mi[len-1]*num[l])%mo;
			sum[top][2]=v1,sum[top][3]=v2;
			v1=(v1+num[l+len]*mi[len])%mo;
			v2=(v2*101+num[l+len])%mo;
			if(v1==v2)
			{
				ans[top]=len+1;
				sum[top][2]=v1,sum[top][3]=v2;
				sum[top][0]=(sum[top][0]+mi[len]*num[r-len])%mo;
				sum[top][1]=(sum[top][1]*101+num[r-len])%mo;
			}
			if(r-l+1>=len+2)
			{
				v1=(v1+num[l+len+1]*mi[len+1]%mo)%mo;
				v2=(v2*101+num[l+len+1])%mo;
				if(v1==v2)
				{
					ans[top]=len+2;
					sum[top][2]=v1,sum[top][3]=v2;
					sum[top][0]=(sum[top][0]+mi[len]*num[r-len])%mo;
					sum[top][1]=(sum[top][1]*101+num[r-len])%mo;
					sum[top][0]=(sum[top][0]+mi[len+1]*num[r-len-1])%mo;
					sum[top][1]=(sum[top][1]*101+num[r-len-1])%mo;
				}	
			}		
		}
		else
		{
			for(;x--;top--)
			{
				if(t[top]=='1') num[r--]=0;
				else num[l++]=0;
				sum[top][0]=sum[top][1]=sum[top][2]=sum[top][3]=0;
			}
		}
		val+=1ll*ans[top];
	}
	printf("%lld\n",val);
}

相关文章: