学了一个下午+一个晚上+一个上午的后缀自动机。。
主要看的是clj的课件以及menci的博客
本来想写学习笔记的,然而要GDOI了,所以之后在写啦!
首先我们知道SAM中一个节点 v v 中长度最小和最大的子串的长度分别为和 max(v) m a x ( v ) ,那么对于一个节点里面子串的数量就是 max(v)−min(v)+1 m a x ( v ) − m i n ( v ) + 1 ,每进来一个数把他加入SAM,ans记录答案即可。
注意注意!!在字符的种类比较大的情况下,比如说int范围内,无法使用一个son数组来保存所有的转移边,那么我们就可以开一个map,花 log l o g 的时间来维护他。
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<map>
using namespace std;
typedef long long ll;
const int N=100010;
inline int read()
{
int x=0,f=1; char ch=getchar();
while(ch<'0' || ch>'9'){if(ch=='-') f=-1; ch=getchar();}
while(ch>='0' && ch<='9'){x=x*10+ch-'0'; ch=getchar();}
return x*f;
}
struct SAM
{
map<int,int> son;
int mx,pre;
SAM(){son.clear(); mx=pre=0;}
}ch[N*2]; int cnt=0,last,start;
int getmin(int x){return ch[ch[x].pre].mx+1;}
int extend(int x)
{
int p=last,np=++cnt;
ch[np].mx=ch[p].mx+1;
while(p && ch[p].son[x]==0) ch[p].son[x]=np,p=ch[p].pre;
if(!p) ch[np].pre=start;
else
{
int q=ch[p].son[x];
if(ch[p].mx+1==ch[q].mx) ch[np].pre=q;
else
{
int nq=++cnt;
ch[nq]=ch[q];
ch[nq].mx=ch[p].mx+1;
ch[q].pre=ch[np].pre=nq;
while(p && ch[p].son[x]==q) ch[p].son[x]=nq,p=ch[p].pre;
}
}
last=np;
return np;
}
int main()
{
int n=read();
ll ans=0; last=start=++cnt;
for(int i=1;i<=n;i++)
{
int x=read();
int v=extend(x);
ans+=(ll)ch[v].mx-getmin(v)+1;
printf("%lld\n",ans);
}
return 0;
}