SDOI2016 生成魔咒

给一个字符串,每次在末尾加入一个字符,询问当前字符串有多少本质不同的子串

裸SAM,因为同一个节点内字符串都是本质不同的,所以每次增加的答案是$mxlen[np] - mxlen[fa[np]]$

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
#define LL long long
const int maxm=5e5+5;
const int inf=2147483233;
int n;char str[maxm];
inline int read()
{
    char ch;int x=0,f=1;
    for(ch=getchar();!isdigit(ch);ch=getchar())if(ch=='-')f=-f;
    for(;isdigit(ch);ch=getchar())x=10*x+ch-'0';
    return x*f;
}
LL ans;
namespace SAM
{
    struct state{int fa,val;map<int,int> ch;}tr[maxn];
    int sz=1,rt=1,lst=1;
    inline void extend(int c)
    {
        int p=lst,np=++sz;
        tr[np].val=tr[p].val+1;
        for(;p&&!tr[p].ch[c];p=tr[p].fa) tr[p].ch[c]=np;
        if(!p) tr[np].fa=rt;
        else
        {
            int q=tr[p].ch[c];
            if(tr[q].val==tr[p].val+1) tr[np].fa=q;
            else
            {
                int nq=++sz;
                tr[nq]=tr[q];tr[nq].val=tr[p].val+1;
                tr[q].fa=tr[np].fa=nq;
                for(;p&&tr[p].ch[c]==q;p=tr[p].fa) tr[p].ch[c]=nq;
            }
        }
        lst=np;
        ans += (LL)tr[np].val - tr[tr[np].fa].val;
    }
}
using namespace SAM;
int rnk[maxn],L,R;
int main()
{
    int ll = read();
    while(ll--)
    {
        int x = read();
        extend(x);
        printf("%lld
",ans);
    }
}
View Code
原文地址:https://www.cnblogs.com/Kong-Ruo/p/9876253.html