bzoj 2555: SubString

题意:

在线增加末尾字符,在线询问子串出现次数。

题解:

lct维护sam的right。
link时把x到根的路径标记一下,cut时也标记一下。
自己YY的竟然能A。
话说我的lct好慢啊。
code:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;
struct SAM{
    int a[27],max,par;
}sam[1200010];int root,tail,tot;
struct trnode{
    int son[2],c,d,fa,rev,u;
}tr[1200010];
void update(int x)
{
    int lc=tr[x].son[0],rc=tr[x].son[1];
    if(tr[x].rev)
    {
        tr[x].rev=0;swap(tr[x].son[0],tr[x].son[1]);
        tr[lc].rev^=1;tr[rc].rev^=1;
    }
    if(tr[x].u!=0)
    {
        int u=tr[x].u;
        tr[x].u=0;
        tr[lc].d+=u;tr[rc].d+=u;
        tr[lc].u+=u;tr[rc].u+=u;
    }
}
bool is_root(int x)
{
    int fa=tr[x].fa;
    if(tr[fa].son[0]==x||tr[fa].son[1]==x) return false;
    return true;
}
int sta[1200010],top;
void pushdown(int x)
{
    top=0;
    while(!is_root(x)) sta[++top]=x,x=tr[x].fa;
    sta[++top]=x;
    while(top) update(sta[top]),top--;
}
void rotate(int x)
{
    int y=tr[x].fa,z=tr[y].fa,R,r,w;
    w=(tr[y].son[0]==x);

    R=y;r=tr[x].son[w];
    tr[R].son[1-w]=r;
    if(r) tr[r].fa=R;

    R=z;r=x;
    if(!is_root(y)) tr[R].son[tr[z].son[1]==y]=r;
    if(r) tr[r].fa=R;

    R=x;r=y;
    tr[R].son[w]=r;
    if(r) tr[r].fa=R;
    update(y);update(x);
}
void splay(int x)
{
    pushdown(x);
    while(!is_root(x))
    {
        int y=tr[x].fa,z=tr[y].fa;
        if(is_root(y)) rotate(x);
        else
            if((tr[y].son[0]==x)==(tr[z].son[0]==y)) rotate(y),rotate(x);
            else rotate(x),rotate(x);
    }
}
void access(int x)
{
    int last=0;
    while(x)
    {
        splay(x);
        tr[x].son[1]=last;
        update(x);
        last=x;x=tr[x].fa;
    }
}
void make_root(int x)
{
    access(x);
    splay(x);
    tr[x].rev^=1;
}
void split(int x,int y)
{
    make_root(x);
    access(y);
    splay(y);
}
void link(int x,int y)
{
    make_root(x);
    tr[x].fa=y;
    split(root,x);
    tr[x].u+=tr[x].d;
}
void cut(int x,int y)
{
    split(root,x);
    tr[x].u-=tr[x].d;
    split(x,y);
    tr[x].fa=tr[y].son[0]=0;
    update(y);
}
int find_root(int x)
{
    access(x);
    splay(x);
    while(tr[x].son[0]) x=tr[x].son[0];
    return x;
}
void addsam(int c)
{
    int p=tail,np=++tot;
    sam[np].max=sam[p].max+1;tr[np].d=1;
    for(;p&&!sam[p].a[c];p=sam[p].par) sam[p].a[c]=np;
    tail=np;
    if(!p) sam[np].par=root,link(np,root);
    else
    {
        int q=sam[p].a[c];
        if(sam[q].max==sam[p].max+1) sam[np].par=q,link(np,q);
        else
        {
            int nq=++tot;sam[nq]=sam[q];
            sam[nq].max=sam[p].max+1;
            link(nq,sam[q].par);
            cut(q,sam[q].par);
            link(q,nq);link(np,nq);
            sam[q].par=sam[np].par=nq;
            for(;p&&sam[p].a[c]==q;p=sam[p].par) sam[p].a[c]=nq;
        }
    }
}
char s1[10],s[3000010];
int main()
{
    root=tail=tot=1;
    int mask=0;
    int T;scanf("%d",&T);
    scanf("%s",s);
    int n=strlen(s);
    for(int i=0;i<n;i++) addsam(s[i]-'A');
    while(T--)
    {
        scanf("%s %s",s1+1,s);
        int n=strlen(s);
        int tmp=mask;
        for(int i=0;i<n;i++)
        {
            tmp=(tmp*131+i)%n;
            swap(s[tmp],s[i]);
        }
        if(s1[1]=='A')
            for(int i=0;i<n;i++) addsam(s[i]-'A');
        else
        {
            int x=root;
            bool ty=false;
            for(int i=0;i<n;i++)
            {
                int c=s[i]-'A';
                if(!sam[x].a[c]){printf("0\n");ty=true;break;}
                x=sam[x].a[c];
            }
            if(!ty)
            {
                make_root(root);
                access(x);splay(x);
                printf("%d\n",tr[x].d);
                mask^=tr[x].d;
            }
        }
    }
}
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值