Splay模板

来源：互联网发布：linux gtk编程编辑：程序博客网时间：2024/05/19 05:01
第二代Splay魔板。
#include <cstdio>#include <cstring>#include <iostream>using namespace std;int fa[100005],ch[100005][2],cnt[100005],size[100005],k[100005],m,ind=0,root=0;void update(int p){    size[0]=0;ch[0][0]=ch[0][1]=0;    size[p]=size[ch[p][0]]+size[ch[p][1]]+cnt[p];}int search(int v){    int p=root;    while(p&&k[p]!=v)p=ch[p][k[p]<v];    return p;}void rotate(int p){    int q=fa[p],y=fa[q],x=ch[q][1]==p;    ch[q][x]=ch[p][x^1];fa[ch[q][x]]=q;    ch[p][x^1]=q;fa[q]=p;fa[p]=y;    if(y)if(ch[y][0]==q)ch[y][0]=p;    else if(ch[y][1]==q)ch[y][1]=p;    update(q);update(p);}void splay(int x){    for(int y;y=fa[x];rotate(x))        if(fa[y])rotate((x==ch[y][0])==(y==ch[fa[y]][0])?y:x);    root=x;}void insert(int v){    if(root){        if(search(v)){            int t=search(v);            splay(t);            size[t]++;cnt[t]++;        }        else{            int x=root,y;            while(1){                y=ch[x][k[x]<v];                if(!y){                    y=++ind;                    k[y]=v;                    ch[y][0]=ch[y][1]=0;                    fa[y]=x;                    ch[x][k[x]<v]=y;                    cnt[y]=1;size[y]=1;                    break;                }                x=y;            }            splay(y);        }    }    else{        root=++ind;        k[root]=v;        ch[root][0]=ch[root][1]=fa[root]=0;        size[root]=1;cnt[root]=1;    }}void remove(int p){    if(!p)return;    splay(p);    if(cnt[p]>1){         cnt[p]--;        size[p]--;        return;    }    int l=ch[p][0],r=ch[p][1];    ch[p][0]=ch[p][1]=0;    if(l)fa[l]=0;    if(r)fa[r]=0;    root=l;    int t=l;    while(t&&ch[t][1])t=ch[t][1];    if(t==0){        root=r;        return;    }    splay(t);    ch[t][1]=r;    fa[r]=t;    update(root);}int grank(int p){    splay(p);    return size[ch[p][0]]+1;}int kth(int _k){    int t=root,k=_k;    while(t){        if(ch[t][0]){            if(size[ch[t][0]]<=k-1&&size[ch[t][0]]+cnt[t]>=k)return t;            if(size[ch[t][0]]>k-1)t=ch[t][0];            else k-=(size[ch[t][0]]+cnt[t]),t=ch[t][1];        }        else{            if(k<=cnt[t])return t;            else k-=cnt[t],t=ch[t][1];        }    }    return 0;}int prefix(int p){    splay(p);    int t=ch[p][0];    while(t&&ch[t][1])t=ch[t][1];    return t;}int suffix(int p){    splay(p);    int t=ch[p][1];    while(t&&ch[t][0])t=ch[t][0];    return t;}int main(){    scanf("%d",&m);    for(int i=0;i<m;i++){        int t1,t2;        scanf("%d%d",&t1,&t2);        switch(t1){            case 1:{                insert(t2);                break;            }            case 2:{                remove(search(t2));                break;            }            case 3:{                printf("%d\n",grank(search(t2)));                break;            }            case 4:{                printf("%d\n",k[kth(t2)]);                break;            }            case 5:{                insert(t2);                printf("%d\n",k[prefix(search(t2))]);                remove(search(t2));                break;            }            case 6:{                insert(t2);                printf("%d\n",k[suffix(search(t2))]);                remove(search(t2));                break;            }        }    }    return 0;}
每次写模板，调试花的时间总是写的时间的三倍以上……写下这行字前觉得可以说的很多，但突然间竟感到渺茫……
阅读全文
0 0