bzoj 4545: DQS的trie 广义后缀自动机+离线+树状数组

题意

DQS的自家阳台上种着一棵颗粒饱满、颜色纯正的trie。
DQS的trie非常的奇特，它初始有n0个节点，n0-1条边，每条边上有一个字符。并且，它拥有极强的生长力：某个i时刻，某个节点就会新生长出一颗子树，它拥有si个节点且节点之间的边上有一个字符，并且新生长出来的子树也是一个树结构。然而因为是新长出来的，根据生活常识可知si必定不会大于i时刻之前的树的大小。
DQS定义trie的子串为从根节点（1号节点）往下走到所有节点所构成的字符串的所有的后缀。DQS身为一个单身doge，常常取出其中一个子串送给妹子，然而他并不希望送给妹子两个相同的子串，所以他非常关心当前trie的本质不同的子串数目。
DQS有时还会去商店购买子串，若他在商店看上某个子串，他希望得知这个子串是否在自家阳台的trie上已经出现，若出现则出现了多少次。如果出现了，他就可以直接回家取trie上的子串辣！
然而DQS身为一个蒟蒻，看着自家阳台的trie树一天天在长大，他被如此众多的节点弄得眼花缭乱，于是他找到了IOI2016Au的你。他会告诉你自家trie树的成长历程，他希望你能够对于每一次询问都做出正确回复。
n是最终树的大小,N<=100000,M<=100000,Si<=当前树的大小

分析

看完题就知道是要在广义sam上搞事。不难想到用lct来维护parents树，但这里有一种我觉得更加优秀的离线做法。
先把整棵trie搞出来然后建sam。
对于询问1，当新加入一个点，那么它到parents树根上的所有没有被算过的点的贡献都要被计算。由于这一定是以1为根的一棵连通子树，所以我们可以暴力往上跳。
对于询问2，实际就是查询该串对应节点的right集大小。用树状数组来维护一下子树中的right集和即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;

typedef long long LL;

const int N=200005;

int n,m,cnt,last[N],id[N],ch[N][3],mx[N],fa[N],c[N],dfn[N],dfn_mx[N],sz,tim,s[N];
LL ans;
struct edge{int to,next,c;}e[N*2];
struct query{int op,s;vector<int> vec;}q[N/2];
char str[N/2];

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;
}

void addedge(int u,int v,int c)
{
    e[++cnt].to=v;e[cnt].c=c;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].c=c;e[cnt].next=last[v];last[v]=cnt;
}

void bit_ins(int x,int y)
{
    while (x<=sz) c[x]+=y,x+=x&(-x);
}

int bit_query(int x)
{
    int ans=0;
    while (x) ans+=c[x],x-=x&(-x);
    return ans;
}

int ins(int ls,int x)
{
    if (ch[ls][x])
    {
        int p=ls,q=ch[p][x];
        if (mx[q]==mx[p]+1) return q;
        int nq=++sz;mx[nq]=mx[p]+1;
        memcpy(ch[nq],ch[q],sizeof(ch[q]));
        fa[nq]=fa[q];fa[q]=nq;
        for (;ch[p][x]==q;p=fa[p]) ch[p][x]=nq;
        return nq;
    }
    int p=ls,np=++sz;mx[np]=mx[p]+1;
    for (;p&&!ch[p][x];p=fa[p]) ch[p][x]=np;
    if (!p) fa[np]=1;
    else
    {
        int q=ch[p][x];
        if (mx[q]==mx[p]+1) fa[np]=q;
        else
        {
            int nq=++sz;mx[nq]=mx[p]+1;
            memcpy(ch[nq],ch[q],sizeof(ch[q]));
            fa[nq]=fa[q];
            fa[q]=fa[np]=nq;
            for (;ch[p][x]==q;p=fa[p]) ch[p][x]=nq;
        }
    }
    return np;
}

void fir_dfs(int x,int fa)
{
    for (int i=last[x];i;i=e[i].next)
    {
        int to=e[i].to;
        if (to==fa) continue;
        id[to]=ins(id[x],e[i].c);
        fir_dfs(to,x);
    }
}

void sec_dfs(int x)
{
    dfn[x]=++tim;
    for (int i=last[x];i;i=e[i].next)
        if (e[i].to!=fa[x]) sec_dfs(e[i].to);
    dfn_mx[x]=tim;
}

void prework()
{
    n=read();n=read();
    for (int i=1;i<n;i++)
    {
        int x=read(),y=read();
        char ch[2];scanf("%s",ch);
        addedge(x,y,ch[0]-'a');
    }
    m=read();
    for (int i=1;i<=m;i++)
    {
        q[i].op=read();
        if (q[i].op==2)
        {
            int r=read(),s=read()-1;q[i].s=s;
            for (int j=1;j<=s;j++)
            {
                int x=read(),y=read();
                char ch[2];scanf("%s",ch);
                addedge(x,y,ch[0]-'a');
            }
        }
        else if (q[i].op==3)
        {
            scanf("%s",str);int len=strlen(str);
            for (int j=0;j<len;j++) q[i].vec.push_back(str[j]-'a');
        }
    }
    sz=id[1]=1;
    fir_dfs(1,0);
    cnt=0;
    for (int i=1;i<=sz;i++) last[i]=0,s[i]=mx[i]-mx[fa[i]];
    for (int i=2;i<=sz;i++) addedge(fa[i],i,0);
    sec_dfs(1);
    for (int i=1;i<=n;i++)
    {
        bit_ins(dfn[id[i]],1);
        int x=id[i];
        while (s[x]) ans+=s[x],s[x]=0,x=fa[x];
    }
}

void solve()
{
    for (int i=1;i<=m;i++)
    {
        if (q[i].op==1) printf("%lld\n",ans);
        else if (q[i].op==2)
        {
            for (int j=n+1;j<=n+q[i].s;j++)
            {
                bit_ins(dfn[id[j]],1);
                int x=id[j];
                while (s[x]) ans+=s[x],s[x]=0,x=fa[x];
            }
            n+=q[i].s;
        }
        else
        {
            int now=1;
            for (int j=0;j<q[i].vec.size();j++) now=ch[now][q[i].vec[j]];
            if (!now) puts("0");
            else printf("%d\n",bit_query(dfn_mx[now])-bit_query(dfn[now]-1));
        }
    }
}

int main()
{
    prework();
    solve();
    return 0;
}