其他分享
首页 > 其他分享> > BZOJ4545 DQS的trie

BZOJ4545 DQS的trie

作者:互联网

Description

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

Solution

对于询问本质不同的子串数目,可以在构建后缀自动机的时候求出

对于询问子串出现的次数,相当于询问该子串对应点在parent树上子树和,其中非复制而来的点权值为1

又因为会有断边操作,所以用LCT维护子树和

数据:链接:https://pan.baidu.com/s/11daDdDrCPJYgwBWJPIgjog  提取码:bzsv

#include<iostream>
#include<cstdio>
using namespace std;
namespace LCT
{
    int tag[200005],tag2[200005],rev[200005],fa[200005],tree[200005][2],v[200005],top,st[200005];
    void pushup(int x)
    {
        tag[x]=tag[tree[x][0]]+tag[tree[x][1]]+tag2[x]+v[x];
    }
    void pushdown(int x)
    {
        if(rev[x])
        {
            if(tree[x][0])
            {
                rev[tree[x][0]]^=1;
                swap(tree[tree[x][0]][0],tree[tree[x][0]][1]);
            }
            if(tree[x][1])
            {
                rev[tree[x][1]]^=1;
                swap(tree[tree[x][1]][0],tree[tree[x][1]][1]);
            }
            rev[x]=0;
        }
    }
    int getson(int x)
    {
        return tree[fa[x]][1]==x;
    }
    int nroot(int x)
    {
        return tree[fa[x]][0]==x||tree[fa[x]][1]==x;
    }
    void rotate(int x)
    {
        int y=fa[x],z=fa[y],b=getson(x),c=getson(y),a=tree[x][!b];
        if(nroot(y))
        {
            tree[z][c]=x;
        }
        fa[x]=z;
        tree[x][!b]=y;
        fa[y]=x;
        tree[y][b]=a;
        if(a)
        {
            fa[a]=y;
        }
        pushup(y);
        pushup(x);
    }
    void splay(int x)
    {
        int top=0;
        st[++top]=x;
        for(int i=x;nroot(i);i=fa[i])
        {
            st[++top]=fa[i];
        }
        while(top)
        {
            pushdown(st[top--]);
        }
        while(nroot(x))
        {
            int y=fa[x];
            if(nroot(y))
            {
                getson(x)^getson(y)?rotate(x):rotate(y);
            }
            rotate(x);
        }
    }
    void access(int x)
    {
        for(int y=0;x;x=fa[y=x])
        {
            splay(x);
            tag2[x]+=tag[tree[x][1]]-tag[y];
            tree[x][1]=y;
            pushup(x);
        }
    }
    void makeroot(int x)
    {
        access(x);
        splay(x);
        rev[x]^=1;
        swap(tree[x][0],tree[x][1]);
    }
    void split(int x,int y)
    {
        makeroot(x);
        access(y);
        splay(y);
    }
    void link(int x,int y)
    {
        split(x,y);
        fa[x]=y;
        tag2[y]+=tag[x];
        pushup(y);
    }
    void cut(int x,int y)
    {
        split(x,y);
        tree[y][0]=fa[x]=0;
        pushup(y);
    }
}
int id,n0,tot,head[100005],m,rt,s,vst[100005],tim,cnt=1,pos[200005];
long long ans;
char str[100005];
struct Edge
{
    int to,nxt,w;
}edge[200005];
struct SAM
{
    int ch[5],fa,len;
}sam[200005];
inline int read()
{
    int w=0,f=1;
    char ch=0;
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        w=(w<<1)+(w<<3)+ch-'0';
        ch=getchar();
    }
    return w*f;
}
int insert(int las,int c)
{
    int p=las,np=++cnt;
    sam[np].len=sam[p].len+1;
    for(;p&&!sam[p].ch[c];p=sam[p].fa)
    {
        sam[p].ch[c]=np;
    }
    if(!p)
    {
        sam[np].fa=1;
    }
    else
    {
        int q=sam[p].ch[c];
        if(sam[q].len==sam[p].len+1)
        {
            sam[np].fa=q;
        }
        else
        {
            int nq=++cnt;
            sam[nq]=sam[q];
            LCT::link(nq,sam[nq].fa);
            sam[nq].len=sam[p].len+1;
            LCT::cut(q,sam[q].fa);
            sam[q].fa=sam[np].fa=nq;
            LCT::link(q,nq);
            for(;p&&sam[p].ch[c]==q;p=sam[p].fa)
            {
                sam[p].ch[c]=nq;
            }
        }
    }
    LCT::tag[np]=LCT::v[np]=1;
    LCT::link(np,sam[np].fa);
    ans+=sam[np].len-sam[sam[np].fa].len;
    return np;
}
void dfs(int k,int f)
{
    for(int i=head[k];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(v!=f&&vst[v]==tim)
        {
            pos[v]=insert(pos[k],edge[i].w);
            dfs(v,k);
        }
    }
}
int main()
{
    id=read();
    n0=read();
    for(int i=1;i<n0;i++)
    {
        int u=read(),v=read();
        char c;
        cin>>c;
        edge[++tot]=(Edge){v,head[u],c-'a'};
        head[u]=tot;
        edge[++tot]=(Edge){u,head[v],c-'a'};
        head[v]=tot;
    }
    pos[1]=1;
    dfs(1,0);
    m=read();
    for(int i=1;i<=m;i++)
    {
        int opt=read();
        if(opt==1)
        {
            printf("%lld\n",ans);
        }
        else if(opt==2)
        {
            ++tim;
            rt=read();
            s=read();
            for(int j=1;j<s;j++)
            {
                int u=read(),v=read();
                char c;
                cin>>c;
                edge[++tot]=(Edge){v,head[u],c-'a'};
                head[u]=tot;
                edge[++tot]=(Edge){u,head[v],c-'a'};
                head[v]=tot;
                vst[u]=vst[v]=tim;
            }
            dfs(rt,0);
        }
        else
        {
            int p=1;
            scanf("%s",str);
            for(int j=0;str[j];j++)
            {
                p=sam[p].ch[str[j]-'a'];
            }
            if(p)
            {
                LCT::makeroot(1);
                LCT::access(p);
                LCT::splay(p);
                printf("%lld\n",LCT::tag2[p]+LCT::tag[LCT::tree[p][1]]+LCT::v[p]);
            }
            else
            {
                puts("0");
            }
        }
    }
    return 0;
}
DQS的trie

 

标签:子串,200005,fa,trie,DQS,tree,int,BZOJ4545
来源: https://www.cnblogs.com/JDFZ-ZZ/p/13542185.html