题目: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 ",ans); return 0; }