[SDOI2016] 生成魔咒 - 后缀自动机

好像写过后缀自动机的题了
直接上板子把XD

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<map>
#define lint long long
#define N 100010
using namespace std;
struct SAM{
    int fa,maxn;
    map<int,int> *ch;
}t[N<<1];int node_cnt;
lint ans;int rt,last;
inline int new_node(int v)
{
    int x=++node_cnt;t[x].maxn=v;
    return t[x].ch=new map<int,int>,x;
}
inline lint extend(int w)
{
    int p=last,np=new_node(t[p].maxn+1);
    while(p&&!t[p].ch->count(w)) (*t[p].ch)[w]=np,p=t[p].fa;
    if(!p) t[np].fa=rt;
    else{
        int q=(*t[p].ch)[w],v=t[p].maxn+1;
        if(t[q].maxn==v) t[np].fa=q;
        else{
            int nq=new_node(v);
            (*t[nq].ch)=(*t[q].ch),t[nq].fa=t[q].fa,t[q].fa=t[np].fa=nq;
            while(p&&t[p].ch->count(w)&&(*t[p].ch)[w]==q)
                (*t[p].ch)[w]=nq,p=t[p].fa;
        }
    }
    return last=np,ans+=t[np].maxn-t[t[np].fa].maxn;
}
int main()
{
    int n;scanf("%d",&n);
    rt=last=new_node(0);
    for(int i=1;i<=n;i++)
    {
        int x;scanf("%d",&x);
        printf("%lld\n",extend(x));
    }
    return 0;
}