【JSOI2008】【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、 询问。语法：Q x y，x, y均为正整数。功能：计算LCQ(x, y) 限制：1 <= x, y <= 当前字符串长度。 2、 修改。语法：R x d，x是正整数，d是字符。功能：将字符串中第x个数修改为字符d。限制：x不超过当前字符串长度。 3、 插入：语法：I x d，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

数据规模：

对于100%的数据，满足：

1、 所有字符串自始至终都只有小写字母构成。

2、 M <= 150,000

3、 字符串长度L自始至终都满足L <= 100,000

4、 询问操作的个数不超过10,000个。

对于第1，2个数据，字符串长度自始至终都不超过1,000

对于第3，4，5个数据，没有插入操作。

Source

开始写这题才发现我已经不会写区间平衡树了..
最后又去看了黄学长的题解..
太久不写区间Splay忘光光..

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 150010
#define U unsigned
#define GET (ch>='0'&&ch<='9')
#define base 31
using namespace std;
int n,m,root,cnt;
int id[MAXN];
char ch[MAXN];
struct Splay
{
    int ch[2],fa,size,val;
    U hash;
}tree[MAXN];
U Pow[MAXN]={1};
void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
void push_up(int x)
{
    int l=tree[x].ch[0],r=tree[x].ch[1];
    tree[x].size=tree[l].size+tree[r].size+1;
    tree[x].hash=tree[l].hash+Pow[tree[l].size]*tree[x].val+tree[r].hash*Pow[tree[l].size+1];
}
void rot(int x,int &f)
{
    int y=tree[x].fa,z=tree[y].fa,l,r;
    l=(tree[y].ch[1]==x);r=l^1;
    if (y==f)   f=x;
    else    tree[z].ch[tree[z].ch[1]==y]=x;
    tree[x].fa=z;tree[y].fa=x;tree[tree[x].ch[r]].fa=y;
    tree[y].ch[l]=tree[x].ch[r];tree[x].ch[r]=y;
    push_up(y);push_up(x);
}
void Splay(int x,int &f)
{
    while (x!=f)
    {
        int y=tree[x].fa,z=tree[y].fa;
        if (y!=f)
        {
            if ((tree[y].ch[0]==x)^(tree[z].ch[0]==y))  rot(x,f);
            else    rot(y,f);
        }
        rot(x,f);
    }
}
int find(int x,int rk)
{
    int l=tree[x].ch[0],r=tree[x].ch[1];
    if (tree[l].size+1==rk) return x;
    if (tree[l].size>=rk)   return find(l,rk);
    return find(r,rk-tree[l].size-1);
}
void insert(int pos,int val)
{
    int x=find(root,pos+1),y=find(root,pos+2);
    Splay(x,root);Splay(y,tree[x].ch[1]);
    int z=++cnt;tree[y].ch[0]=z;tree[z].fa=y;tree[z].val=val;
    push_up(z);push_up(y);push_up(x);
}
U query(int pos,int len)
{
    int x=find(root,pos),y=find(root,pos+len+1);
    Splay(x,root);Splay(y,tree[x].ch[1]);
    int z=tree[y].ch[0];return tree[z].hash;
}
int Query(int x,int y)
{
    int l=1,r=cnt-max(x,y)-1,ret=0,mid=0;
    while (l<=r)
    {
        mid=(l+r)>>1;
        if (query(x,mid)==query(y,mid)) l=mid+1,ret=mid;
        else    r=mid-1;
    }
    return ret;
}
void build(int l,int r,int f)
{
    if (l>r)    return;
    int last=id[f],mid=(l+r)>>1,now=id[mid];
    if (l==r)
    {
        now=id[l];tree[now].val=tree[now].hash=ch[l]-'a'+1;
        tree[now].size=1;tree[now].fa=last;tree[last].ch[l>=f]=now;
        return;
    }
    build(l,mid-1,mid);build(mid+1,r,mid);
    tree[now].val=ch[mid]-'a'+1;tree[now].fa=last;push_up(now);
    tree[last].ch[mid>=f]=now;
}
int main()
{
    scanf("%s",ch+2);n=strlen(ch+2);
    for (int i=1;i<=150004;i++) Pow[i]=Pow[i-1]*base;
    for (int i=1;i<=n+2;i++)    id[i]=i;
    build(1,n+2,0);cnt=n+2;root=(n+3)>>1;
    in(m);int x,y;char opt[3],c[3];
    while (m--)
    {
        scanf("%s",opt+1);in(x);
        if (opt[1]=='Q')    in(y),printf("%d\n",Query(x,y));
        if (opt[1]=='R')    scanf("%s",c+1),x=find(root,x+1),Splay(x,root),tree[root].val=c[1]-'a'+1,push_up(root);
        if (opt[1]=='I')    scanf("%s",c+1),insert(x,c[1]-'a'+1);
    }
}