1. 程式人生 > >【Luogu3804】【模板】後綴自動機(後綴自動機)

【Luogu3804】【模板】後綴自動機(後綴自動機)

longest lib amp .so fin end 自動 mes 累加

【Luogu3804】【模板】後綴自動機(後綴自動機)

題面

洛谷

題解

一個串的出現次數等於\(right/endpos\)集合的大小
而這個集合的大小等於所有\(parent\)樹上兒子的大小
這樣子的話,給每個終止位置的\(size\)記為\(1\)
然後按照拓撲序累加,這就是\(right/endpos\)集合的大小
最後對於每個\(size>1\)的節點,\(ans=max(longest*size)\)

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath> #include<algorithm> #include<set> #include<map> #include<vector> #include<queue> using namespace std; #define ll long long #define RG register #define MAX 2500000 inline int read() { RG int x=0,t=1;RG char ch=getchar(); while((ch<'0'||ch>'9'
)&&ch!='-')ch=getchar(); if(ch=='-')t=-1,ch=getchar(); while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar(); return x*t; } int last=1,num=1; char ch[1111111]; ll ans; int size[MAX],c[MAX],a[MAX]; struct Node { int son[26]; int ff,len; }t[MAX]; void
extend(int c) { int p=last,np=++num;last=np; t[np].len=t[p].len+1; while(p&&!t[p].son[c])t[p].son[c]=np,p=t[p].ff; if(!p)t[np].ff=1; else { int q=t[p].son[c]; if(t[p].len+1==t[q].len)t[np].ff=q; else { int nq=++num; t[nq]=t[q]; t[nq].len=t[p].len+1; t[q].ff=t[np].ff=nq; while(p&&t[p].son[c]==q)t[p].son[c]=nq,p=t[p].ff; } } size[np]=1; } int main() { scanf("%s",ch+1); for(int i=1,l=strlen(ch+1);i<=l;++i)extend(ch[i]-97); for(int i=1;i<=num;++i)c[t[i].len]++; for(int i=1;i<=num;++i)c[i]+=c[i-1]; for(int i=1;i<=num;++i)a[c[t[i].len]--]=i; for(int i=num;i;--i) { int now=a[i]; size[t[now].ff]+=size[now]; if(size[now]>1)ans=max(ans,1ll*size[now]*t[now].len); } printf("%lld\n",ans); return 0; }

【Luogu3804】【模板】後綴自動機(後綴自動機)