BZOJ1014: [JSOI2008]火星人prefix
Description
火星人最近研究了一种操作:求一个字串两个后缀的公共前缀。
比方说,有这样一个字符串:madamimadam,我们将这个字符串的各个字符予以标号:
序号: 1 2 3 4 5 6 7 8 9 10 11
字符: m a d a m i m a d a m
现在,火星人定义了一个函数LCQ(x, y),表示:该字符串中第x个字符开始的字串,与该字符串中第y个字符开始的字串,两个字串的公共前缀的长度。
比方说,LCQ(1, 7) = 5, LCQ(2, 10) = 1, LCQ(4, 7) = 0
在研究LCQ函数的过程中,火星人发现了这样的一个关联:
如果把该字符串的所有后缀排好序,就可以很快地求出LCQ函数的值;同样,如果求出了LCQ函数的值,也可以很快地将该字符串的后缀排好序。
尽管火星人聪明地找到了求取LCQ函数的快速算法,但不甘心认输的地球人又给火星人出了个难题:
在求取LCQ函数的同时,还可以改变字符串本身。
具体地说,可以更改字符串中某一个字符的值,也可以在字符串中的某一个位置插入一个字符。
地球人想考验一下,在如此复杂的问题中,火星人是否还能够做到很快地求取LCQ函数的值。
Input
第一行给出初始的字符串。
第二行是一个非负整数M,表示操作的个数。
接下来的M行,每行描述一个操作。
操作有3种,如下所示:
1、询问。语法:Q x y,x,y均为正整数。功能:计算LCQ(x,y)限制:1<=x,y<=当前字符串长度。
2、修改。语法:R x d,x是正整数,d是字符。功能:将字符串中第x个数修改为字符d。限制:x不超过当前字符串长度。
3、插入:语法:I x d,x是非负整数,d是字符。功能:在字符串第x个字符之后插入字符d,如果x=0,则在字符串开头插入。限制:x不超过当前字符串长度
Output
对于输入文件中每一个询问操作,你都应该输出对应的答案。一个答案一行。
Sample Input
7
Q 1 7
Q 4 8
Q 10 11
R 3 a
Q 1 7
I 10 a
Q 2 11
Sample Output
1
0
2
1
HINT
1、所有字符串自始至终都只有小写字母构成。
2、M<=150,000
3、字符串长度L自始至终都满足L<=100,000
4、询问操作的个数不超过10,000个。
对于第1,2个数据,字符串长度自始至终都不超过1,000
对于第3,4,5个数据,没有插入操作。
题解Here!
这个题,难到掉牙了。。。
看到字符串处理,坑能会想到 AC自动机/ SA / SAM 等等。
但是看到有插入与修改操作, Splay 啊!
然后维护一个 hash 值:
a[rt].hash=a[lson].hash*val[a[rson].s+1]+val[a[rson].s]*(a[rt].v-'a'+1)+a[rson].hash; //lson是rt的左孩子,rson是rt的右孩子 //a[rt].v表示rt节点存的字母 //a[rt].s表示rt节点的子树大小 //val[i]表示第i位的hash值,即27^i
剩下的就是 Splay 基本操作,具体可见BZOJ1500: [NOI2005]维修数列
附代码:
#include<iostream> #include<algorithm> #include<cstdio> #include<cstring> #define MAXN 100010 using namespace std; int n,m; int root=0,size=1; char ch[MAXN]; unsigned val[MAXN]; struct Splay{ int f,s,son[2]; char v; unsigned hash; }a[MAXN]; inline int read(){ int date=0,w=1;char c=0; while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();} while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();} return date*w; } inline void pushup(int rt){ if(!rt)return; int lson=a[rt].son[0],rson=a[rt].son[1]; a[rt].s=a[lson].s+a[rson].s+1; a[rt].hash=a[lson].hash*val[a[rson].s+1]+val[a[rson].s]*(a[rt].v-'a'+1)+a[rson].hash; } inline void turn(int rt,int k){ int x=a[rt].f,y=a[x].f; a[x].son[k^1]=a[rt].son[k]; if(a[rt].son[k])a[a[rt].son[k]].f=x; a[rt].f=y; if(y)a[y].son[a[y].son[1]==x]=rt; a[x].f=rt; a[rt].son[k]=x; pushup(x);pushup(rt); } void splay(int rt,int ancestry){ while(a[rt].f!=ancestry){ int x=a[rt].f,y=a[x].f; if(y==ancestry)turn(rt,a[x].son[0]==rt); else{ int k=a[y].son[0]==x?1:0; if(a[x].son[k]==rt){turn(rt,k^1);turn(rt,k);} else{turn(x,k);turn(rt,k);} } } if(ancestry==0)root=rt; } inline int newnode(char x){ int rt=size++; a[rt].v=x;a[rt].hash=x-'a'+1; a[rt].s=1; a[rt].son[0]=a[rt].son[1]=0; return rt; } int buildtree(int l,int r){ if(l>r)return 0; int mid=l+r>>1,lson=0,rson=0; lson=buildtree(l,mid-1); int rt=newnode(ch[mid]); rson=buildtree(mid+1,r); a[rt].son[0]=lson; a[rt].son[1]=rson; if(lson)a[lson].f=rt; if(rson)a[rson].f=rt; pushup(rt); return rt; } int kth(int rt,int k){ while(1){ int y=a[rt].son[0]; if(k>a[y].s+1){ k-=a[y].s+1; rt=a[rt].son[1]; } else if(k<=a[y].s)rt=y; else return rt; } } inline int split(int l,int r){ int front=kth(root,l),next=kth(root,r); splay(front,0);splay(next,front); return a[next].son[0]; } inline void change(int x,char y){ int front=kth(root,x),next=kth(root,x+2); splay(front,0);splay(next,front); int u=a[next].son[0]; a[u].v=y;a[u].hash=y-'a'+1; pushup(next);pushup(front); } inline void insert(int x,char y){ int front=kth(root,x),next=kth(root,x+1); splay(front,0);splay(next,front); int rt=newnode(y); a[next].son[0]=rt;a[rt].f=next; pushup(next);pushup(front); } inline bool check(int x,int y,int len){ int u=split(x,x+len+1),hu=a[u].hash; int v=split(y,y+len+1),hv=a[v].hash; return (hu==hv); } void work(){ char f[2],y[2]; int x,k; while(m--){ scanf("%s",f);x=read(); if(f[0]=='Q'){ k=read(); int l=1,r=min(size-x,size-k)-2,mid; while(l<=r){ mid=l+r>>1; if(check(x,k,mid))l=mid+1; else r=mid-1; } printf("%d\n",r); } if(f[0]=='R'){ scanf("%s",y); change(x,y[0]); } if(f[0]=='I'){ scanf("%s",y); insert(x+1,y[0]); } } } void init(){ scanf("%s",ch+1); n=strlen(ch+1); val[0]=1; for(int i=1;i<=MAXN-5;i++)val[i]=val[i-1]*27; root=buildtree(0,n+1); m=read(); } int main(){ init(); work(); return 0; }