BZOJ 1014 [JSOI2008]火星人prefix

Posted 2020-10-08 Achen

splay维护hash值；

看到大佬们都在打数据结构，我说好的不打数据结构又自己打脸了。

为了写这个昨天还特意去打了Splay的普通平衡树，自从我学会Treap以来第一次用splayA掉普通平衡树QAQ

没有在前后加点，所以特判了在前后加字母的情况。查询就二分答案，然后转过去取hash值，判断是否相等。

一开始用了mod 1e9+7，狂T不止，到处翻别人的博客听说用unsigned long long 自然溢出很快，改后就快了一倍。。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#include<ctime>
#define lc ch[x][0]
#define rc ch[x][1]
typedef unsigned long long LL;
const int maxn=300000+299;
using namespace std;
int len,root,tot,T,x,y,sz[maxn],ch[maxn][2],p[maxn];
LL v[maxn],ha[maxn],base=131,power[maxn];
char s[maxn],op[10],c[10];
inline int read() {
    char ch=getchar(); int f=1,ret=0;
    while(ch!=‘-‘&&(ch>‘9‘||ch<‘0‘)) ch=getchar();
    if(ch==‘-‘) f=-1,ch=getchar();
    for(;ch>=‘0‘&&ch<=‘9‘;ch=getchar()) ret=ret*10+ch-‘0‘;
    return ret*f;
}
void update(int x) {
    sz[x]=sz[lc]+sz[rc]+1;
    ha[x]=(ha[lc]*power[sz[rc]+1]+v[x]*(power[sz[rc]])+ha[rc]);
}
void rotate(int x) {
    int y=p[x],z=p[y],l=(x==ch[y][1]),r=l^1;
    if(z) ch[z][y==ch[z][1]]=x; p[x]=z;
    ch[y][l]=ch[x][r]; p[ch[x][r]]=y;
    ch[x][r]=y; p[y]=x;
    update(y); update(x);
}
void splay(int x,int FA) {
    if(p[x]==FA) return;
    for(;p[x]!=FA;rotate(x)) {
        int y=p[x],z=p[y];
        if(z!=FA) (y==ch[z][1])^(x==ch[y][1])?rotate(x):rotate(y);
    }
    if(!FA) root=x;
}
int build(int l,int r) {
    int x=++tot;
    sz[x]==1;
    int mid=(l+r)>>1;
    v[x]=s[mid-1];
    if(mid-1>=l) lc=build(l,mid-1);
    if(mid+1<=r) rc=build(mid+1,r);
    if(lc) p[lc]=x;
    if(rc) p[rc]=x;
    update(x);
    return x;
}
int kth(int k) {
    int res=-1;
    for(int x=root;x;) {
        if(sz[lc]+1==k) return x;
        if(sz[lc]+1>k) x=lc;
        else k-=(sz[lc]+1),x=rc;
    }
}
LL find(int l,int r) {
    int ll=kth(l),rr=kth(r);
    if(ll==rr) return v[ll];
    splay(ll,0);
    splay(rr,ll);
    LL res=v[root]*power[sz[ch[ch[root][1]][0]]+1]+ha[ch[ch[root][1]][0]]*power[1]+v[ch[root][1]];
    return res;
}
int ef(int x,int y) {
    int res=0;
    if(x>y) swap(x,y);
    if(x==y) return sz[root]-y+1;
    int l=0,r=sz[root]-y;
    while(l<=r) {
        int mid=(l+r)>>1;
        if(find(x,x+mid)==find(y,y+mid)) res=mid+1,l=mid+1;
        else r=mid-1;
    }
    return res;
}
int main()
{
    scanf("%s",s);
    len=strlen(s); 
    power[0]=1;
    for(int i=1;i<=maxn-290;i++) power[i]=power[i-1]*base;
    root=build(1,len);
    T=read();
    while(T--) {
        scanf("%s",op);
        if(op[0]==‘Q‘) {
            scanf("%d%d",&x,&y);
            printf("%d\n",ef(x,y));
        }
        else if(op[0]==‘R‘) {
            scanf("%d%s",&x,c);
            int ll=kth(x);
            splay(ll,0);
            v[root]=c[0];
            update(root);
        }
        else {
            scanf("%d%s",&x,c);
            if(x==0) {
                int ll=kth(1);
                splay(ll,0);
                ch[root][0]=++tot;
                sz[tot]=1; p[tot]=root;
                v[tot]=ha[tot]=c[0];
                update(root);
            }
            else {
                int ll=kth(x);
                splay(ll,0);
                if(x==sz[root]) {
                    ch[root][1]=++tot;
                    sz[tot]=1; p[tot]=root;
                    v[tot]=ha[tot]=c[0];
                    update(root);
                }
                else {
                    int ll=kth(x);
                    splay(ll,0);
                    int rr=kth(x+1);
                    splay(rr,ll);
                    ch[rr][0]=++tot;
                    sz[tot]=1; p[tot]=rr;
                    v[tot]=ha[tot]=c[0];
                    update(rr);
                    update(root);
                }
            }
        }
    }
    return 0;
}

View Code