又是一道 SAM 神题.
Code:
#include<bits/stdc++.h>
#define maxn 400002
#define setIO(s) freopen(s".in","r",stdin)
using namespace std;
namespace tr
{
#define lson t[x].l
#define rson t[x].r
#define mid ((l+r)>>1)
int cnt;
struct Node { int l,r; }t[maxn*20];
void modify(int &x,int l,int r,int k)
{
if(!x) x=++cnt;
if(l==r) return;
if(k<=mid) modify(lson,l,mid,k);
else modify(rson,mid+1,r,k);
}
int merge(int u,int v)
{
if(!u||!v) return u+v;
int x=++cnt;
lson=merge(t[u].l,t[v].l);
rson=merge(t[u].r,t[v].r);
return x;
}
int query(int x,int l,int r,int L,int R)
{
if(!x) return 0;
if(l>=L&&r<=R) return 1;
int tmp=0;
if(L<=mid) tmp+=query(lson,l,mid,L,R);
if(R>mid) tmp+=query(rson,mid+1,r,L,R);
return tmp;
}
};
namespace SAM
{
char str[maxn];
int last,tot,n;
int trans[maxn][27],f[maxn],len[maxn],C[maxn],rk[maxn],top[maxn],dp[maxn],pos[maxn],rt[maxn];
void init() { last=tot=1; }
void extend(int c,int i)
{
int np=++tot,p=last;
len[np]=len[p]+1,last=np;
while(p&&!trans[p][c]) trans[p][c]=np,p=f[p];
if(!p) f[np]=1;
else
{
int q=trans[p][c];
if(len[q]==len[p]+1) f[np]=q;
else
{
int nq=++tot;
len[nq]=len[p]+1;
pos[nq]=pos[q];
memcpy(trans[nq],trans[q],sizeof(trans[q]));
f[nq]=f[q], f[np]=f[q]=nq;
while(p&&trans[p][c]==q) trans[p][c]=nq,p=f[p];
}
}
pos[np]=i;
tr::modify(rt[last],1,n,i);
}
void prepare()
{
int i,j;
init();
scanf("%d%s",&n,str+1);
for(i=1;i<=n;++i) extend(str[i]-'a',i);
for(i=1;i<=tot;++i) ++C[len[i]];
for(i=1;i<=tot;++i) C[i]+=C[i-1];
for(i=1;i<=tot;++i) rk[C[len[i]]--]=i;
for(i=tot;i>1;--i)
{
int u=rk[i];
rt[f[u]]=tr::merge(rt[f[u]],rt[u]);
}
}
void calc()
{
int i,j,ans=1;
for(i=2;i<=tot;++i)
{
int u=rk[i],ff=f[rk[i]];
if(ff==1) { top[u]=u,dp[u]=1; continue; }
if(tr::query(rt[top[ff]],1,n,pos[u]-len[u]+len[top[ff]],pos[u]-1))
top[u]=u,dp[u]=dp[top[ff]]+1;
else
top[u]=top[ff];
ans=max(ans,dp[u]);
}
printf("%d\n",ans);
}
};
int main()
{
//msetIO("input");
SAM::prepare();
SAM::calc();
return 0;
}