1. 程式人生 > >洛谷 P3804 [模板] 字尾自動機

洛谷 P3804 [模板] 字尾自動機

題目:https://www.luogu.org/problemnew/show/P3804

模仿了一篇題解,感覺很好寫啊。

程式碼如下:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const xn=2e6+5;
int n,lst=1,cnt=1,go[xn][30],fa[xn],l[xn],siz[xn],tax[xn],a[xn];
char s[xn];
void add(int w)
{
  
int p=lst,np=++cnt; lst=np; l[np]=l[p]+1; for(;p&&!go[p][w];p=fa[p])go[p][w]=np; if(!p)fa[np]=1;//1 else { int q=go[p][w]; if(l[q]==l[p]+1)fa[np]=q; else { int nq=++cnt; l[nq]=l[p]+1; fa[nq]=fa[q]; fa[q]=nq; fa[np]=nq; memcpy(go[nq],go[q],sizeof
go[q]); for(;go[p][w]==q;p=fa[p])go[p][w]=nq; } } siz[np]=1; } int main() { scanf("%s",s+1); n=strlen(s+1); for(int i=1;i<=n;i++)add(s[i]-'a'); for(int i=1;i<=cnt;i++)tax[l[i]]++; for(int i=1;i<=cnt;i++)tax[i]+=tax[i-1]; for(int i=1;i<=cnt;i++)a[tax[l[i]]--]=i; ll ans
=0; for(int i=cnt;i;i--) { int p=a[i]; siz[fa[p]]+=siz[p]; if(siz[p]>1)ans=max(ans,(ll)siz[p]*l[p]); } printf("%lld\n",ans); return 0; }