SPOJ LCS2 Longest Common Substring II ——后缀自动机

后缀自动机裸题

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define F(i,j,k) for (int i=j;i<=k;++i)
#define D(i,j,k) for (int i=j;i<=k;--i)
#define inf 0x3f3f3f3f
#define ll long long
#define maxn 200005

struct sam{
char s[maxn];
int h[maxn],to[maxn],ne[maxn],en,ttt;
int dp[maxn][10],n;
int last,cnt,len,lth;
int go[maxn][26],l[maxn],fa[maxn];
void addedge(int a,int b)
{to[en]=b;ne[en]=h[a];h[a]=en++;}
void init()
{
memset(h,-1,sizeof h);en=0;
last=cnt=1;
//        memset(go,0,sizeof go);
//        memset(dp,0,sizeof dp);
}
void add(int x)
{
//      printf("add %d\n",x);
int p=last,np=last=++cnt; l[np]=l
+1;
//      printf("on point %d\n",np);
for (;p&&!go[p][x];p=fa[p]) go[p][x]=np;
if (!p) fa[np]=1;
else
{
int q=go[p][x];
if (l[q]==l[p]+1) fa[np]=q;
else
{
int nq=++cnt;
l[nq]=l[p]+1;
memcpy(go[nq],go[q],sizeof go[q]);
fa[nq]=fa[q];
fa[np]=fa[q]=nq;
for (;p&&go[p][x]==q;p=fa[p]) go[p][x]=nq;
}
}
}
void dfs(int o)
{
for (int i=h[o];i>=0;i=ne[i])
{
dfs(to[i]);
F(j,1,ttt)
dp[o][j]=max(dp[o][j],dp[to[i]][j]);
}
}
void solve()
{
scanf("%d",&n);
scanf("%s",s+1);
lth=len=strlen(s+1);
F(i,1,lth) add(s[i]-'a');
int i=0;
while (scanf("%s",s+1)!=EOF)
{
++i;
//            scanf("%s",s+1);
len=strlen(s+1);
int now=1,t=0;
F(j,1,len)
{
int x=s[j]-'a';
if (go[now][x]) {t++;now=go[now][x];}
else
{
while (now&&!go[now][x]) now=fa[now];
if (!now) {t=0;now=1;}
else t=l[now]+1,now=go[now][x];
}
dp[now][i]=max(dp[now][i],t);
//                for (int k=now;k;k=fa[k]) dp[k][i]=max(dp[k][i],t);
}
}
dfs(1);
int ans=0;ttt=i;
F(i,1,cnt) addedge(fa[i],i);
F(i,1,cnt)
{
int tmp=l[i];
F(j,2,ttt)
tmp=min(tmp,dp[i][j]);
ans=max(ans,tmp);
}
printf("%d\n",ans);
}
}SAM;

int main()
{
SAM.init();
SAM.solve();
}