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对于输入文件中每一个询问操作,你都应该输出对应的答案。一个答案一行。
用平衡树维护哈希值求LCP。
二分答案logn,查询nlogn,总时间复杂度O(nlog2n)。
#include<cstdio>
#include<cstring>
struct node
{
char c;
int k[2],f,s;
unsigned long long h;
}t[110000];
int max(int x,int y)
{
return x>y?x:y;
}
int min(int x,int y)
{
return x<y?x:y;
}
int p=13331,s,r;
char ss[110000];
unsigned long long pow[110000];
void up(int p)
{
t[p].s=t[t[p].k[0]].s+t[t[p].k[1]].s+1;
t[p].h=t[t[p].k[0]].h*pow[t[t[p].k[1]].s+1]+(t[p].c-'a'+1)*pow[t[t[p].k[1]].s]+t[t[p].k[1]].h;
}
void rot(int p,bool b)
{
int x=t[p].k[b];
int y=t[x].k[!b];
int z=t[p].f;
t[p].k[b]=y;
if (y) t[y].f=p;
t[x].k[!b]=p;
t[p].f=x;
t[x].f=z;
if (z) t[z].k[t[z].k[1]==p]=x;
up(p);
up(x);
}
void splay(int p)
{
int x,y,z;
while (t[p].f)
{
x=t[p].f;
if (!t[x].f)
{
rot(x,t[x].k[1]==p);
return;
}
y=t[x].f;
if ((p==t[x].k[0])==(x==t[y].k[0]))
{
rot(y,t[y].k[1]==x);
rot(x,t[x].k[1]==p);
}
else
{
rot(x,t[x].k[1]==p);
rot(y,t[y].k[1]==p);
}
}
}
int build(int l,int r)
{
if (l>r) return 0;
int p=++s,x=build(l,(l+r)/2-1),y=build((l+r)/2+1,r);
t[p].c=ss[(l+r)/2];
t[p].k[0]=x;
t[p].k[1]=y;
if (x) t[x].f=p;
if (y) t[y].f=p;
up(p);
return p;
}
void init()
{
scanf("%s",ss+1);
int l=strlen(ss+1);
ss[0]=ss[l+1]=0;
r=build(0,l+1);
}
int find(int p)
{
int x=r;
while (1)
{
if (p<=t[t[x].k[0]].s) x=t[x].k[0];
else
{
if (p==t[t[x].k[0]].s+1)
{
splay(x);
return x;
}
else
{
p-=(t[t[x].k[0]].s+1);
x=t[x].k[1];
}
}
}
}
void ins(int p,char c)
{
int x,y,z,w,f;
char c1,c2;
bool b;
r=x=find(p);
y=t[x].k[1];
while (t[y].k[0]) y=t[y].k[0];
t[y].k[0]=w=++s;
t[w].f=y;
t[w].s=1;
t[w].c=c;
t[w].h=c-'a'+1;
z=y;
while (y) up(y),y=t[y].f;
splay(w);
r=w;
}
void mod(int p,char c)
{
int x,y,z;
r=x=find(p);
t[x].c=c;
up(x);
}
unsigned long long hash(int p,int q)
{
r=find(p-1);
t[t[r].k[1]].f=0;
int y=find(q+1);
unsigned long long ans=t[t[y].k[0]].h;
t[y].f=r;
t[r].k[1]=y;
up(r);
return ans;
}
int que(int x,int y)
{
int z,w,p,q,mid;
unsigned long long h1,h2;
p=0;
q=s-max(x,y);
while (p<q)
{
mid=(p+q+1)/2;
h1=hash(x,x+mid-1);
h2=hash(y,y+mid-1);
if (h1==h2) p=mid;
else q=mid-1;
}
return p;
}
int main()
{
int i,j,k,x,y,z,m;
char s[5],c;
pow[0]=1;
for (i=1;i<=105000;i++)
pow[i]=pow[i-1]*p;
init();
scanf("%d",&m);
for (i=1;i<=m;i++)
{
scanf("%s",s+1);
if (s[1]=='Q')
scanf("%d%d",&x,&y),printf("%d\n",que(x+1,y+1));
if (s[1]=='R')
scanf("%d%*c%c",&x,&c),mod(x+1,c);
if (s[1]=='I')
scanf("%d%*c%c",&x,&c),ins(x+1,c);
}
}