【bzoj2258】【文本校对】【splay+hash】

Description

 

为了给Wind买生日礼物，Jiajia不得不找了一份检查文本的工作。这份工作很无聊：给你一段文本 
要求比对从文本中某两个位置开始能匹配的最大长度是多少。但比无聊更糟糕的是，Jiajia的经理 
还可能往文本里面插入一些字符。 
Jiajia想用一个程序来解决这些繁琐的工作。这个程序的速度要足够快，因为Wind的生日就快要到了 
Jiajia必须赚到足够多的钱，也就是处理足够多的文本。 

Input

输入文件第一行是原始文本。 
输入文件第二行是操作数n。此后n行，每行描述一条命令，命令有两种形式： 
I ch p：表示将一个字符ch插入到当前文本的第p个字符之前，如果p大于当前文本长度则表示插入到当前文本末尾； 
Q i j：表示询问当前文本从原始文本的第i个和第j个字符现在所在的位置开始能匹配的字符是多少。 
你可以认为文本初始长度不超过50000，I命令最多200条，Q命令最多20000条。 

Output

对于每条Q命令输出一行，为最长匹配长度。 

Sample Input

abaab
5
Q 1 2
Q 1 3
I a 2
Q 1 2
Q 1 3

Sample Output

0
1
0
3

题解：splay+hash;记录一下原序列中每个数在平衡树中节点的编号即可。

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 1000010
#define ll unsigned int
using namespace std;
int c[N][3],size[N],rt,id[N],n,m,fa[N],sz,x,y;
char ch[N],s[10],ss[10];
ll h[N],v[N],p[N];
inline void updata(int x){
 size[x]=size[c[x][1]]+size[c[x][0]]+1;
 h[x]=h[c[x][0]]+v[x]*p[size[c[x][0]]]+p[size[c[x][0]]+1]*h[c[x][1]];
}
inline int read(){
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
inline void rotate(int x,int &k){
 int y=fa[x],z=fa[y],l,r;
 if (c[y][0]==x) l=0;else l=1;r=l^1;
 if (y==k) k=x;
 else{
    if (c[z][0]==y) c[z][0]=x;
    else c[z][1]=x;              
  }
  fa[x]=z;fa[c[x][r]]=y;fa[y]=x;
  c[y][l]=c[x][r];c[x][r]=y;  
  updata(y);updata(x);   
}
inline void splya(int x,int &k){
  while (x!=k){
    int y=fa[x],z=fa[y];
    if (y!=k){
      if (c[z][0]==y^c[y][0]==x) rotate(x,k);
      else rotate(y,k); 
    }
    rotate(x,k);
  } 
}
int find(int k,int rk){
  if (size[c[k][0]]+1==rk) return k;
  if (size[c[k][0]]>=rk) return find(c[k][0],rk);
  else return find(c[k][1],rk-size[c[k][0]]-1);
}
inline int query(int a,int k){
  int x=find(rt,a),y=find(rt,a+k+1);
  splya(x,rt);splya(y,c[x][1]);
  return h[c[y][0]];
}
inline int solve(int x,int y){
 int l=1,r=min(sz-x,sz-y)-1,ans=0;
 while(l<=r){
  int mid=(l+r)>>1;
  if(query(x,mid)==query(y,mid))l=mid+1,ans=mid;
  else r=mid-1;
  }
  return ans;
}
inline void insert(int l,int val){
  int x=find(rt,l+1),y=find(rt,l+2);
  splya(x,rt);splya(y,c[x][1]);
  c[y][0]=++sz;fa[sz]=y;size[sz]=1;v[sz]=val;
  updata(sz);updata(y);updata(x);
}
void build(int l,int r,int f){
  if (l>r) return;
  int now=id[l],last=id[f];
  if (l==r){
    v[now]=h[now]=ch[l]-'a'+1;
    fa[now]=last;size[now]=1;
    if (l<f) c[last][0]=now;
    else c[last][1]=now;  
    return;
  }
  int mid=(l+r)>>1;now=id[mid];
  build(l,mid-1,mid);build(mid+1,r,mid);
  fa[now]=last;v[now]=ch[mid]-'a'+1;updata(now);
  if (mid<f) c[last][0]=now;else c[last][1]=now;
  updata(now);
}
int main(){
  scanf("%s",ch+2);p[0]=1;n=strlen(ch+2);
  for (int i=1;i<=150004;i++) p[i]=p[i-1]*151;
  for (int i=1;i<=n+2;i++) id[i]=i;  
  build(1,n+2,0);rt=(n+3)>>1;sz=n+2;
  m=read();
  for (int i=1;i<=m;i++){
    scanf("%s",s+1);
    if (s[1]=='Q'){
      x=read();y=read();x++;y++;
	  splya(x,rt);x=size[c[x][0]]+1;
	  splya(y,rt);y=size[c[y][0]]+1;
	  printf("%d\n",solve(x-1,y-1));
	} 
    if (s[1]=='I'){scanf("%s",ss+1);x=read();insert(x-1,ss[1]-'a'+1);}
 }
}