回文自动机模板题
头铁了一下午hdu6599,最后发现自己的板有问题
先放这里一个正确性得到基本确认的板,过两天肝hdu6599
#pragma GCC optimize(2) #include<bits/stdc++.h> #include<iostream> #include<cstring> #include<cassert> #define MAXN 300010 #define LL long long #define BASE 2LL #define MOD 1000000007 using namespace std; char s[MAXN]; int len; LL ans[MAXN]; int qpow(int base,int n){ LL ans=1; while(n){ if(n&1)ans=(ans*base)%MOD; base=(1LL*base*base)%MOD; n>>=1; } return ans; } struct PTnode{ int len,fail,son[26]; LL cnt;//该点对应的回文串出现次数 //建完了树,要再跑一遍,cnt才是正确的 PTnode(){ cnt=len=fail=0; memset(son,0,sizeof son); } }PTdian[MAXN<<1]; int PTlast,PTnum; int PTgetfail(int i,int x){ while(s[i-PTdian[x].len-1]!=s[i]) { x=PTdian[x].fail; } return x; } void PTextend(int i,int x){ int cur=PTgetfail(i,PTlast); if(!PTdian[cur].son[x]){ int now=++PTnum; PTdian[now].len=PTdian[cur].len+2; PTdian[now].fail=PTdian[PTgetfail(i,PTdian[cur].fail)].son[x]; PTdian[cur].son[x]=now; } PTdian[PTdian[cur].son[x]].cnt++; PTlast=PTdian[cur].son[x]; } void PTcount(){ for(int i=PTnum;i>=2;i--){ //逆序累加 PTdian[PTdian[i].fail].cnt+=PTdian[i].cnt; } } //int dfs(int l,int index,int hash1,int hash2){ // if(index>1 && hash1==hash2){ // ans[PTdian[index].len]+=PTdian[index].cnt; // } // for(int i=0;i<26;i++){ // if(PTdian[index].son[i]!=0){ // LL tmp1=(hash1+i*qpow(BASE,l)%MOD)%MOD; // LL tmp2=(BASE*hash2%MOD+i)%MOD; // dfs(l+1,PTdian[index].son[i],tmp1,tmp2); // } // } //} //char a[MAXN]; //int dfs1(int depth,int index){ // //检查回文自动机构造正确性 // for(int i=0;i<depth;i++)printf("%c",a[i]); // printf(" len:%d siz:%d\n",PTdian[index].len,PTdian[index].cnt); // for(int i=0;i<26;i++){ // if(PTdian[index].son[i]!=0){ // a[depth]=i+‘a‘; // dfs1(depth+1,PTdian[index].son[i]); // } // } //} int main(){ // freopen("1.txt","r",stdin); // freopen("2.txt","w",stdout); while(1){ int rep=scanf("%s",s); if(rep==EOF)break; len=strlen(s); memset(ans,0,sizeof ans);memset(PTdian,0,sizeof PTdian); PTlast=PTnum=1; PTdian[1].len=-1; PTdian[0].fail=PTdian[1].fail=1; for(int i=0;i<len;i++){ PTextend(i,s[i]-‘a‘); } PTcount(); LL maxx=0; for(int i=2;i<=PTnum;i++){ maxx=max(maxx,1LL*PTdian[i].cnt*PTdian[i].len); } printf("%lld\n",maxx); // dfs(0,0,0,0); // dfs(0,1,0,0); // dfs1(0,0); // dfs1(0,1); // for(int i=1;i<=len;i++)printf("%lld%c",ans[i],(i!=len)?‘ ‘:‘\n‘); } }
原文地址:https://www.cnblogs.com/isakovsky/p/11261531.html
时间: 2024-11-07 13:25:54