【JSOI2008】火星人prefix 哈希 非旋转treap

题目大意

　　就是给你一个字符串，有三种操作，共\(m\)个

　　　\(Q~x~y\)：询问第\(x\)个后缀和第\(y\)个后缀的LCP

　　　\(R~x~y\)：把第\(x\)个字符改成\(y\)

　　　\(I~x~y\)：在第\(x\)个字符后面插入一个字符\(y\)

　　\(m\leq 150000,\)任何时候字符串长度\(\leq 100000\)，询问个数\(\leq 10000\)

题解

　　直接用平衡树维护哈希值，询问时二分答案。我用unsigned long long存回TLE，用unsigned存就能过。

　　时间复杂度：\(O(q\log^2n+m\log n)\)（\(q\)为询问个数）

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<ctime>
#include<utility>
using namespace std;
typedef long long ll;
typedef unsigned ui;
typedef pair<int,int> pii;
struct node
{
	int v;
	int k;
	int ls,rs;
	int sz;
	ui s;
	node()
	{
		v=k=ls=rs=sz=0;
		s=0;
	}
};
node a[1000010];
int cnt=0;
ui pw[1000010];
int newnode(int v)
{
	cnt++;
	a[cnt].v=v;
	a[cnt].k=rand();
	a[cnt].s=v;
	a[cnt].sz=1;
	return cnt;
}
void mt(int p)
{
	a[p].s=a[a[p].ls].s*pw[a[a[p].rs].sz+1]+a[p].v*pw[a[a[p].rs].sz]+a[a[p].rs].s;
	a[p].sz=a[a[p].ls].sz+a[a[p].rs].sz+1;
}
int merge(int x,int y)
{
	if(!x||!y)
		return x+y;
	if(a[x].k<a[y].k)
	{
		a[x].rs=merge(a[x].rs,y);
		mt(x);
		return x;
	}
	else
	{
		a[y].ls=merge(x,a[y].ls);
		mt(y);
		return y;
	}
}
pii split(int x,int v)
{
	if(!x)
		return pii();
	if(v<=a[a[x].ls].sz)
	{
		pii s=split(a[x].ls,v);
		a[x].ls=s.second;
		s.second=x;
		mt(x);
		return s;
	}
	else
	{
		pii s=split(a[x].rs,v-a[a[x].ls].sz-1);
		a[x].rs=s.first;
		s.first=x;
		mt(x);
		return s;
	}
}
char s[1000010];
int rt=0;
int n;
ui gethash(int l,int r)
{
	if(l>r)
		return 0;
	pii s1=split(rt,r);
	pii s2=split(s1.first,l-1);
	ui res=a[s2.second].s;
	s1.first=merge(s2.first,s2.second);
	rt=merge(s1.first,s1.second);
	return res;
}
int query(int x,int y)
{
	int l=0;
	int r=min(n-x+1,n-y+1);
	while(l<r)
	{
		int mid=(l+r+1)>>1;
		if(gethash(x,x+mid-1)==gethash(y,y+mid-1))
			l=mid;
		else
			r=mid-1;
	}
	return l;
}
void insert(int x,int v)
{
	pii s1=split(rt,x);
	s1.first=merge(s1.first,newnode(v));
	rt=merge(s1.first,s1.second);
	n++;
}
void change(int x,int v)
{
	pii s1=split(rt,x);
	pii s2=split(s1.first,x-1);
	a[s2.second].s=a[s2.second].v=v;
	s1.first=merge(s2.first,s2.second);
	rt=merge(s1.first,s1.second);
}
int main()
{
	freopen("bzoj1014.in","r",stdin);
	freopen("bzoj1014.out","w",stdout);
	srand(12700);
	int i,m;
	pw[0]=1;
	for(i=1;i<=1000000;i++)
		pw[i]=pw[i-1]*127;
	scanf("%s",s+1);
	n=strlen(s+1);
	for(i=1;i<=n;i++)
		rt=merge(rt,newnode(s[i]-'a'+1));
	scanf("%d",&m);
	int x,y;
	for(i=1;i<=m;i++)
	{
		scanf("%s",s+1);
		if(s[1]=='Q')
		{
			scanf("%d%d",&x,&y);
			printf("%d\n",query(x,y));
		}
		else if(s[1]=='R')
		{
			scanf("%d%s",&x,s+1);
			change(x,s[1]-'a'+1);
		}
		else
		{
			scanf("%d%s",&x,s+1);
			insert(x,s[1]-'a'+1);
		}
	}
	return 0;
}