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、询问。语法:Qxy,x,y均为正整数。功能:计算LCQ(x,y)限制:1<=x,y<=当前字符串长度。
2、修改。语法:Rxd,x是正整数,d是字符。功能:将字符串中第x个数修改为字符d。限制:x不超过当前字
符串长度。
3、插入:语法:Ixd,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个数据,没有插入操作。
Solution
这题很容易想到用Splay来解决。每个Splay的区间维护它所对应的字符串的哈希,然后二分答案判断。复杂度mlog^2n。
注意:
1.二分答案的时候要注意当前答案不能超出串长,要打特判;
2.注意输入;
3.如果在头尾插虚点方便操作,要注意询问LCQ(x,x)的时候答案会比正确答案+1,要打特判。
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef unsigned long long LL;
const int maxn=100010,seed=53;
struct node{
int fa,ch[2],size;
LL hash;
char s;
}T[maxn];
int n,m,num=0,top=0,h[256],root;
char ch[maxn];
LL pow[maxn];
void create(char c){
T[++num].s=c;
T[num].size=1;
T[num].ch[0]=T[num].ch[1]=0;
T[num].hash=h[c];
}
void update(int x){
T[x].size=1;T[x].hash=0;
if(T[x].ch[0]){
T[x].size+=T[T[x].ch[0]].size;
T[x].hash+=T[T[x].ch[0]].hash;
}
T[x].hash+=h[T[x].s]*pow[T[T[x].ch[0]].size];
if(T[x].ch[1]){
T[x].size+=T[T[x].ch[1]].size;
T[x].hash+=T[T[x].ch[1]].hash*pow[T[T[x].ch[0]].size+1];
}
}
int build(int l,int r,int fa){
if(l>r)return 0;
int mid=(l+r)>>1;
create(ch[mid]);
int temp=num;
T[temp].fa=fa;
T[temp].ch[0]=build(l,mid-1,temp);
T[temp].ch[1]=build(mid+1,r,temp);
update(temp);
return temp;
}
int getson(int p){
return p==T[T[p].fa].ch[1];
}
void rotate(int p){
if(!T[p].fa)return;
int k=getson(p),fa=T[p].fa;
int fafa=T[fa].fa;
T[fa].ch[k]=T[p].ch[k^1];
if(T[p].ch[k^1])T[T[p].ch[k^1]].fa=fa;
T[p].ch[k^1]=fa;
T[fa].fa=p;
T[p].fa=fafa;
if(fafa)T[fafa].ch[fa==T[fafa].ch[1]]=p;
update(fa);update(p);
}
void Splay(int u,int f){
for(int fa;(fa=T[u].fa)!=f;rotate(u)){
if(T[fa].fa!=f){
rotate((getson(u)==getson(fa))?fa:u);
}
}
if(!f)root=u;
}
int Rank(int x){
int p=root;
while(1){
if(x<=T[T[p].ch[0]].size)p=T[p].ch[0];
else{
int temp=T[T[p].ch[0]].size+1;
if(x==temp)return p;
x-=temp;
p=T[p].ch[1];
}
}
}
void Insert(int l,char c){
int x=Rank(l),y=Rank(l+1);
Splay(x,0);Splay(y,x);
create(c);
T[y].ch[0]=num;
T[num].fa=y;
update(y);update(x);
}
void Change(int l,char c){
int x=Rank(l);
Splay(x,0);
T[x].s=c;
update(x);
}
bool check(int l,int r,int ans){
int x=Rank(l-1),y=Rank(l+ans);
Splay(x,0);Splay(y,x);
LL temp=T[T[y].ch[0]].hash;
x=Rank(r-1),y=Rank(r+ans);
Splay(x,0);Splay(y,x);
return temp==T[T[y].ch[0]].hash;
}
int Query(int x,int y){
if(x>y)swap(x,y);
int l=0,r=T[root].size-1,mid;
while(l+1<r){
mid=(l+r)>>1;
if(y+mid>T[root].size){
r=mid;
continue;
}
if(check(x,y,mid))l=mid;
else r=mid;
}
if(y+r<=T[root].size&&check(x,y,r))return r;
else return l;
}
void Init(){
ch[0]=' ';
scanf("%s%d",ch+1,&m);
n=strlen(ch);
ch[n]=' ';
n=strlen(ch);
T[0].size=T[0].hash=T[0].fa=T[0].ch[0]=T[0].ch[1]=0;
pow[0]=1;h[' ']=52;
for(int i=1;i<=maxn-10;i++)
pow[i]=pow[i-1]*seed;
for(int i='A';i<='Z';i++)
h[i]=top++;
for(int i='a';i<='z';i++)
h[i]=top++;
root=build(0,n-1,0);
}
void Work(){
char opt[10],c[10];
int x,y;
while(m--){
scanf("%s%d",opt,&x);
if(opt[0]=='Q'){
scanf("%d",&y);
if(x==y)printf("%d\n",T[root].size-x-1);
else printf("%d\n",Query(x+1,y+1));
}
else if(opt[0]=='R'){
scanf("%s",c);
Change(x+1,c[0]);
}
else{
scanf("%s",c);
Insert(x+1,c[0]);
}
}
}
int main(){
Init();
Work();
return 0;
}