AC日记——[JSOI2008]火星人prefix bzoj 1014

1014

思路：

　　平衡树+二分答案+hash；

　　好了懂了吧。

代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

#define maxn 2000005

int ch[maxn][],key[maxn],f[maxn],root,size[maxn];
int n,m,tot;

long long ha[maxn],mi[maxn];

char ai[maxn];

inline void in(int &now)
{
    char Cget=getchar();now=;
    while(Cget>''||Cget<'') Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}

inline bool getson(int now)
{
    return ch[f[now]][]==now;
}

inline void updata(int now)
{
    size[now]=,ha[now]=;
    if(ch[now][]) size[now]+=size[ch[now][]],ha[now]+=ha[ch[now][]];
    ha[now]+=key[now]*mi[size[now]-];
    if(ch[now][]) ha[now]+=ha[ch[now][]]*mi[size[now]],size[now]+=size[ch[now][]];
}

inline void rotate(int now,int &to)
{
    int fa=f[now],ffa=f[fa];bool pos=getson(now);
    ch[fa][pos]=ch[now][pos^];
    if(ch[fa][pos]) f[ch[fa][pos]]=fa;
    if(to==fa) to=now;
    else ch[ffa][getson(fa)]=now;
    ch[now][pos^]=fa,f[fa]=now,f[now]=ffa;
    updata(fa),updata(now);
}

void splay(int now,int &to)
{
    while(now!=to)
    {
        int fa=f[now],ffa=f[fa];
        if(fa!=to)
        {
            if(ch[fa][]==now^ch[ffa][]==fa) rotate(now,to);
            else rotate(now,to);
        }rotate(now,to);
    }
}

void find(int p,bool pos)
{
    int now=root;
    while()
    {
        if(size[ch[now][]]>=p) now=ch[now][];
        else
        {
            p-=size[ch[now][]];
            if(p==)
            {
                if(pos) splay(now,root);
                else splay(now,ch[root][]);
                return ;
            }
            else p--,now=ch[now][];
        }
    }
}

void insert(int p,int ci)
{
    find(p,true),find(p+,false);
    ch[ch[root][]][]=++tot;
    key[tot]=ci,size[tot]=,f[tot]=ch[root][],ha[tot]=ci;
    updata(ch[root][]),updata(root);
}

int tree_build(int l,int r,int fa)
{
    int now=l+r>>;key[now]=ai[now],f[now]=fa;
    if(now>l) ch[now][]=tree_build(l,now-,now);
    if(now<r) ch[now][]=tree_build(now+,r,now);
    updata(now);return now;
}

long long hash_ci(int p,int len)
{
    find(p,true),find(p+len+,false);
    return ha[ch[ch[root][]][]];
}

int main()
{
    mi[]=;
    for(int i=;i<=;i++) mi[i]=mi[i-]*27LL;
    scanf("%s",ai+),tot=strlen(ai+)+;
    for(int i=;i<=tot-;i++) ai[i]=ai[i]-'a'+;
    root=tree_build(,tot,);
    in(m);char op[];int u,v;
    for(;m--;)
    {
        scanf("%s",op);
        if(op[]=='Q')
        {
            in(u),in(v);
            if(u>v) swap(u,v);
            int l=,r=tot-v-,ans=;
            while(l<=r)
            {
                int mid=l+r>>;
                if(hash_ci(u,mid)==hash_ci(v,mid)) ans=mid,l=mid+;
                else r=mid-;
            }
            printf("%d\n",ans);
        }
        else if(op[]=='R')
        {
            in(u),scanf("%s",op);
            find(u+,true),key[root]=op[]-'a'+,updata(root);
        }
        else
        {
            in(u),scanf("%s",op);
            insert(u+,(int)(op[]-'a'+));
        }
    }
    return ;
}