BZOJ 4516: [Sdoi2016]生成魔咒

后缀自动机的基础应用：计算不同的子串个数。我们只需要将每个点的max_len-min_len+1的和维护一下就行了，这里要引进一个结论，就是一个节点的min_len==其parent节点的max_len+1，具体怎么证我就不说了，总之代入之后就将公式变为了this->max_len - this->parent->max_len,这样就可以在建后缀自动机的时候顺道维护出来答案了。还有，这题由于给的数字不知道是多少，所以son指针要用map维护

#include<cstdio>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include<cstring>
#include<string>
#include<iostream>
#include<iomanip>
#include<algorithm>
#include<map>
using namespace std;
struct sam
{
    map<int,sam*> son;
    sam* parent;
    int max_len;
    sam(int _=0):max_len(_),parent(0x0){}
}*root=new sam(),*last=root;
long long ans=0;
void my_insert(int x)
{
    sam *p=last;
    sam *np=new sam(p->max_len+1);
    while(p && !p->son[x])
    {
        p->son[x]=np;
        p=p->parent;
    }
    if(!p) np->parent=root;
    else
    {
        sam *q=p->son[x];
        if(q->max_len==p->max_len+1) np->parent=q;
        else
        {
            sam *nq=new sam(p->max_len+1);
            ans-=(q->max_len-q->parent->max_len);
            nq->parent=q->parent;
            nq->son=q->son;
            q->parent=nq,np->parent=nq;
            while(p && p->son[x]==q)
            {
                p->son[x]=nq;
                p=p->parent;
            }
            ans+=(q->max_len-q->parent->max_len);
            ans+=(nq->max_len-nq->parent->max_len);
        }
    }
    last=np;
    ans+=(np->max_len-np->parent->max_len);
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        my_insert(x);
        printf("%lld\n",ans);
    }
    return 0;
}