【清华集训2017模拟12.09】塔
题目
有一个塔,他的名字叫做粽粑,粽粑的每一层都有一个颜色 .
粽粑非常厉害,它在吸收天地精华之后会长高.粽粑的长高方式有两种:
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);
}
