bzoj 2555: SubString

Posted 2020-11-04 哈

显然，在后缀自动机的后缀树上插入一个后缀后，此时表示整串的节点(np)到根的所有节点，其表示的串作为子串的出现次数都要+1；发现需要链修改，动态连边/删边，因此用LCT维护

注意：虽然没明确写，但根据讨论区以及做题情况，此题字符集只有AB两个字母

错误记录：没有msk^=res

  1 #include<cstdio>
  2 #include<algorithm>
  3 #include<cstring>
  4 using namespace std;
  5 int Add(int a,int b)    {return a+b;}
  6 int Add(int a,int b,int c)    {return a+b+c;}
  7 int Mul(int a,int b)    {return a*b;}
  8 namespace LCT
  9 {
 10 const int N=1200000;
 11 struct Node
 12 {
 13     Node *ch[2],*fa;
 14     bool rev;
 15     int addv;
 16     int dat,sz;
 17     void upd()    {sz=(ch[0]?ch[0]->sz:0)+1+(ch[1]?ch[1]->sz:0);}
 18     void padd(int x)    {addv=Add(addv,x);dat=Add(dat,x);}
 19     void pd()
 20     {
 21         if(rev)
 22         {
 23             swap(ch[0],ch[1]);
 24             if(ch[0])    ch[0]->rev^=1;
 25             if(ch[1])    ch[1]->rev^=1;
 26             rev=0;
 27         }
 28         if(addv)
 29         {
 30             if(ch[0])    ch[0]->padd(addv);
 31             if(ch[1])    ch[1]->padd(addv);
 32             addv=0;
 33         }
 34     }
 35     bool isroot()    {return (!fa)||(this!=fa->ch[0]&&this!=fa->ch[1]);}
 36     bool gson()    {return this==fa->ch[1];}
 37     void rot()//将自身向上旋，要求已经完成下传标记
 38     {
 39         bool d=gson();Node *f=fa;
 40         fa=f->fa;if(!f->isroot())    f->fa->ch[f->gson()]=this;
 41         f->ch[d]=ch[!d];if(ch[!d])    ch[!d]->fa=f;
 42         f->fa=this;ch[!d]=f;
 43         f->upd();upd();
 44     }
 45 }nodes[N+100];
 46 Node *st[N+100];int top;
 47 int mem;
 48 Node *getnode()
 49 {
 50     Node *t=nodes+mem++;t->sz=1;//t->ch[0]=t->ch[1]=t->fa=0;t->rev=t->addv=t->dat=0;
 51     return t;
 52 }
 53 void solvetag(Node *o)
 54 {
 55     while(!o->isroot())    st[++top]=o,o=o->fa;
 56     st[++top]=o;
 57     while(top)    st[top--]->pd();
 58 }
 59 void splay(Node *o)
 60 {
 61     solvetag(o);
 62     for(;!o->isroot();o->rot())
 63         if(!o->fa->isroot())
 64             o->gson()==o->fa->gson()?o->fa->rot():o->rot();
 65 }
 66 void acc(Node *o)
 67 {
 68     for(Node *lst=0;o;lst=o,o=o->fa)
 69         splay(o),o->ch[1]=lst,o->upd();
 70 }
 71 void mtop(Node *o)    {acc(o);splay(o);o->rev^=1;}
 72 void link(Node *x,Node *y)    {acc(x);splay(x);x->fa=y;}//保证x是根，使x成为y子节点
 73 void cut(Node *x)    {acc(x);splay(x);x->ch[0]->fa=0;x->ch[0]=0;}//断开x与父亲
 74 void add(Node *y,int t)    {acc(y);splay(y);y->padd(t);}//y到根路径+t
 75 int query(Node *y)    {splay(y);return y->dat;}//查询y单点的权值
 76 }
 77 using LCT::Node;using LCT::link;using LCT::getnode;using LCT::cut;using LCT::query;
 78 namespace SAM
 79 {
 80 int mem,np,root;
 81 int len[1200100],fa[1200100],trans[1200100][3];
 82 Node *nd[1200100];
 83 void append(int c)
 84 {
 85     int p=np;np=++mem;len[np]=len[p]+1;nd[np]=getnode();
 86     for(;p&&trans[p][c]==0;p=fa[p])    trans[p][c]=np;
 87     if(!p)    link(nd[np],nd[root]),fa[np]=root;
 88     else
 89     {
 90         int q=trans[p][c];
 91         if(len[q]==len[p]+1)    link(nd[np],nd[q]),fa[np]=q;
 92         else
 93         {
 94             int nq=++mem;nd[nq]=getnode();nd[nq]->dat=query(nd[q]);
 95             cut(nd[q]);link(nd[nq],nd[fa[q]]);link(nd[q],nd[nq]);link(nd[np],nd[nq]);
 96             fa[nq]=fa[q];fa[q]=fa[np]=nq;
 97             memcpy(trans[nq],trans[q],sizeof(trans[nq]));len[nq]=len[p]+1;
 98             for(;p&&trans[p][c]==q;p=fa[p])    trans[p][c]=nq;
 99         }
100     }
101     LCT::add(nd[np],1);
102 }
103 int query(const char *ss,int len)
104 {
105     int now=root,i;
106     for(i=0;i<len;i++)
107     {
108         now=trans[now][ss[i]-‘A‘];
109         if(!now)    return 0;
110     }
111     return query(nd[now]);
112 }
113 }
114 char tmp[3000100],ttt[10];int len,q;
115 void decode(int msk)
116 {
117     for(int j=0;j<len;j++)
118     {
119         msk=(msk*131+j)%len;
120         swap(tmp[j],tmp[msk]);
121     }
122 }
123 int main()
124 {
125     int msk=0,res,i;SAM::np=SAM::root=++SAM::mem;SAM::nd[SAM::root]=getnode();
126     scanf("%d",&q);
127     scanf("%s",tmp);len=strlen(tmp);
128     for(i=0;i<len;i++)    SAM::append(tmp[i]-‘A‘);
129     while(q--)
130     {
131         scanf("%s",ttt);
132         if(ttt[0]==‘A‘)
133         {
134             scanf("%s",tmp);len=strlen(tmp);decode(msk);
135             for(i=0;i<len;i++)    SAM::append(tmp[i]-‘A‘);
136         }
137         else if(ttt[0]==‘Q‘)
138         {
139             scanf("%s",tmp);len=strlen(tmp);decode(msk);
140             res=SAM::query(tmp,len);msk^=res;printf("%d\n",res);
141         }
142     }
143     return 0;
144 }