洛谷 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
",ans);
  return 0;
}
原文地址:https://www.cnblogs.com/Zinn/p/10105883.html