[BZOJ4516][Sdoi2016]生成魔咒(后缀自动机)

传送门


学了一个下午+一个晚上+一个上午的后缀自动机。。
主要看的是clj的课件以及menci的博客
本来想写学习笔记的,然而要GDOI了,所以之后在写啦!

首先我们知道SAM中一个节点 v 中长度最小和最大的子串的长度分别为 m i n ( v ) m a x ( v ) ,那么对于一个节点里面子串的数量就是 m a x ( v ) m i n ( v ) + 1 ,每进来一个数把他加入SAM,ans记录答案即可。

注意注意!!在字符的种类比较大的情况下,比如说int范围内,无法使用一个son数组来保存所有的转移边,那么我们就可以开一个map,花 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;
}

猜你喜欢

转载自blog.csdn.net/cabi_zgx/article/details/80081323