[题解]bzoj1014 火星人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、询问。语法: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

madamimadam
7
Q 1 7
Q 4 8
Q 10 11
R 3 a
Q 1 7
I 10 a
Q 2 11

Sample Output

5
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;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值