当初学 Splay树 所看过的，凭理解打过的代码

 

 

HDU 1890

 

这是我第一次接触Splay树时，所接触的代码。

那位大牛的博客我现在找不到了#。#。抱歉ing..

 

思路就是区间翻转，主要是将n个数通过树的先序访问保存下来。

这样旋转的时候就不会出现问题

 

代码：

 

//
//  1890.cpp
//  ACM_HDU
//
//  Created by ipqhjjybj on 13-8-27.
//  Copyright (c) 2013年 ipqhjjybj. All rights reserved.
//  参考着别人Ac代码敲的
//
#include <cstdio>
#include <algorithm>
#include <map>
using namespace std;
struct SplayTree{
    const static int maxn=111111;
    int n,tot,root;
    map<int,int> Map;
    int child[maxn][2];
    int pre[maxn],size[maxn],flex[maxn];
    int id[maxn];
    struct node{
        int id,val;
        bool operator < (const node & tmp) const{
            return val<tmp.val||(val==tmp.val&&id<tmp.id);
        }
    }in[maxn];
    void Pushup(int &x){
        size[x]=size[child[x][0]]+size[child[x][1]]+1;
    }
    void Pushdown(int &x){
        if(flex[x]){
            flex[x]=0;
            flex[child[x][0]]^=1;
            flex[child[x][1]]^=1;
            swap(child[x][0],child[x][1]);
        }
    }
    void Del_root(){
        int t=root;
        if(child[root][1]){
            root=child[root][1];
            Select(1,0);
            child[root][0]=child[t][0];
            if(child[t][0]) pre[child[t][0]]=root;
        }else
            root=child[root][0];
        pre[root]=0;
        Pushup(root);
    }
    // c==0  左旋 ,  c==1 右旋
    inline void Rotate(int x, int c) {    // 旋转， c=0 左旋， c=1 右旋
        int y = pre[x];
        Pushdown(y);
        Pushdown(x);
        child[y][!c] = child[x][c];
        if ( child[x][c] )    pre[ child[x][c] ] = y;
        pre[x] = pre[y];
        if ( pre[y] )    child[ pre[y] ][ child[pre[y]][1] == y ] = x;
        child[x][c] = y;
        pre[y] = x;
        Pushup(y);
    }
    void Splay(int x,int f){
       // puts("fuck Splay");
        Pushdown(x);
        while(pre[x]!=f){
        //    printf("f=%d pre[%d]=%d\n",f,x,pre[x]);
            int y=pre[x],z=pre[y];
            Pushdown(z),Pushdown(y),Pushdown(x);
            if(pre[pre[x]]==f){
                Rotate(x,child[pre[x]][0]==x);
            }else{
                if ( child[z][0] == y ) {
                    if ( child[y][0] == x )
                        Rotate(y, 1), Rotate(x, 1);
                    else
                        Rotate(x, 0), Rotate(x, 1);
                }
                else {
                    if ( child[y][0] == x )
                        Rotate(x, 1), Rotate(x, 0);
                    else
                        Rotate(y, 0), Rotate(x, 0);
                }
            }
        }
        Pushup(x);
        if(f==0) root=x;
    }
    void Select(int k,int f){
        int x=root;
        while(1){
            Pushdown(x);
            if(k==size[child[x][0]]+1)
                break;
            if(k<=size[child[x][0]])
                x=child[x][0];
            else{
                k-=size[child[x][0]]+1;
                x=child[x][1];
            }
        }
        Splay(x,f);
    }
    void Newnode(int &x,int f){
        //puts("fuck newnode");
        x=++tot;
        child[x][0]=child[x][1]=0;
        pre[x]=f;
        flex[x]=0;
        size[x]=1;
    }
    void Build(int &x,int l,int r,int f){
        if(l>r)return;
        int mid=(l+r)>>1;
        //puts("Fuck(Build)");
        Newnode(x,f);
        Map[id[mid]]=x;
        Build(child[x][0],l,mid-1,x);
        Build(child[x][1],mid+1,r,x);
        Pushup(x);
    }
    void init(int _n){
        n=_n; Map.clear(); root=tot=0;
        pre[0]=child[0][0]=child[0][1]=0;
        size[0]=flex[0]=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&in[i].val);
            in[i].id=i;
        }
        sort(in+1,in+1+n);
        for(int i=1;i<=n;i++)
            id[in[i].id]=i;
        Map[id[1]]=1;
        Map[id[n]]=2;
        Newnode(root,0);
        Newnode(child[root][1],root);
        Build(child[child[root][1]][0],2,n-1,child[root][1]);
        Pushup(child[root][1]);
        Pushup(root);
    }
    void solve(int n){
        for(int i=1;i<=n;i++){
            //printf("i=%d\n",i);
            //printf("Map[%d]=%d\n",i,Map[i]);
            Splay(Map[i],0);
            //printf("root=%d\n",root);
            printf("%d%c",i+size[child[root][0]],i==n?'\n':' ');
            flex[child[root][0]]^=1;
            Del_root();
        }
    }
}spt;
int main(){
    int n;
    while(scanf("%d",&n)&&n){
        if(n==1){scanf("%*d");puts("1");continue;}
        spt.init(n);
        spt.solve(n);
    }
    return 0;
}

 

 

 

HDU 3487

 

别人都说这是水题。。

设计到Splay的基本操作，区间翻转及区间插入。。

对[l,r]这段区间，要想获得的话，先将l-1这个节点翻转到根节点，再把r+1这个节点翻转到根节点之下。

这样KeyTree  child[child[root][1]][0] 所表示的那段就是区间[l,r]了。 之后再进行区间的删除跟插入。

找到要插入的区间，比如插到i之后，就先把i旋转到根节点，然后然后把i的右子树中的最小的那个点旋转到根下面，这样这个最小点的左节点就是空的节点，可以插入那段区间了。

翻转的操作反而更加简单， 只用获得节点，传递一个修改的标记的就好了。

 

http://blog.csdn.net/acm_cxlove/article/details/7795244 原博客

代码：

//
//  3487.cpp
//  ACM_HDU
//
//  Created by ipqhjjybj on 13-9-6.
//  Copyright (c) 2013年 ipqhjjybj. All rights reserved.
//  先看完别人代码。后来凭记忆敲的Ac代码
//
#include<iostream>
#include<cstring>
#include<queue>
#include<cstdio>
#include<algorithm>
#define N 300015
#define inf 1<<29
#define MOD 100000007
#define LL long long
#define _match(a,b) ((a)==(b))

#define Key_value ch[ch[root][1]][0]

using namespace std;
int n,q;
int size[N],pre[N],key[N],num[N],rev[N];
int ch[N][2],tot,root,node[N];
void Push_Up(int &t){
    size[t]=size[ch[t][0]]+size[ch[t][1]]+1;
}
void Push_Down(int &t){
    if(rev[t]){
        swap(ch[t][0],ch[t][1]);
        rev[ch[t][0]]^=1;
        rev[ch[t][1]]^=1;
        rev[t]=0;
    }
}
void NewNode(int &t,int k,int father){
    t=++tot;
    pre[t]=father;
    key[t]=k;
    rev[t]=0;
    ch[t][0]=ch[t][1]=0;
}
void Build(int &t,int l,int r,int father){
    if(l>r)return;
    int mid=(l+r)>>1;
    NewNode(t,mid,father);
    Build(ch[t][0],l,mid-1,t);
    Build(ch[t][1],mid+1,r,t);
    Push_Up(t);
}
void Init(){
    root=tot=0;
    ch[root][0]=ch[root][1]=pre[root]=rev[root]=size[root]=0;
    NewNode(root,-1,0);
    NewNode(ch[root][1],-1,root);
    size[root]=2;
    Build(Key_value,1,n,ch[root][1]);
    Push_Up(ch[root][1]);
    Push_Up(root);
}
void Rotate(int x,int c){ // c==0左旋，c==1右旋
    int y=pre[x];
    Push_Down(y);
    Push_Down(x);
    ch[y][!c]=ch[x][c];
    pre[ch[x][c]]=y;
    if(pre[y])
        ch[pre[y]][ch[pre[y]][1]==y]=x;
    pre[x]=pre[y];
    ch[x][c]=y;
    pre[y]=x;
    Push_Up(y);
}
void Splay(int x,int father){
    Push_Down(x);
    while(pre[x]!=father){
        int y=pre[x],z=pre[y];
        Push_Down(z),Push_Down(y),Push_Down(z);
        if(pre[y]==father){
            Rotate(x,ch[y][0]==x);
        }else{
            if(ch[z][0]==y){
                if(ch[y][0]==x)
                    Rotate(y,1),Rotate(x,1);
                else Rotate(x,0),Rotate(x,1);
            }else{
                if(ch[y][0]==x)
                    Rotate(x,1),Rotate(x,0);
                else Rotate(y,0),Rotate(x,0);
            }
        }
    }
    Push_Up(x);
    if(father==0) root=x;
}
int Get_Kth(int t,int k){
    Push_Down(t);
    if(size[ch[t][0]]==k-1)
        return t;
    else if(size[ch[t][0]]>=k)
        return Get_Kth(ch[t][0],k);
    else return Get_Kth(ch[t][1],k-size[ch[t][0]]-1);
}
int Min_Value(int t){
    Push_Down(t);
    while(ch[t][0]){
        t=ch[t][0];
        Push_Down(t);
    }
    return t;
}
void Reversal(int a,int b){
    a=Get_Kth(root,a);
    b=Get_Kth(root,b+2);
    Splay(a,0);
    Splay(b,root);
    rev[Key_value]^=1;
}
void Cut(int a,int b,int c){
    int x=Get_Kth(root,a);
    int y=Get_Kth(root,b+2);
    Splay(x,0);
    Splay(y,root);
    int tmp=Key_value;
    Key_value=0;
    Push_Up(ch[root][1]);
    Push_Up(root);
    int cm=Get_Kth(root,c+1);
    Splay(cm,0);
    int mi=Min_Value(ch[root][1]);
    Splay(mi,root);
    Key_value=tmp;
    pre[Key_value]=ch[root][1];
    Push_Up(ch[root][1]);
    Push_Up(root);
}
int cnt;
void InOrder(int r){
    if(r==0)
        return;
    Push_Down(r);
    InOrder(ch[r][0]);
    if(cnt>=1&&cnt<=n){
        if(cnt>1)printf(" ");
        printf("%d",key[r]);
    }
    cnt++;
    InOrder(ch[r][1]);
}
int main(){
    while(scanf("%d%d",&n,&q)!=EOF){
		if(n==-1&&q==-1)
			break;
		Init();
		while(q--){
			char str[10];
			int a,b,c;
			scanf("%s",str);
			if(str[0]=='C'){
				scanf("%d%d%d",&a,&b,&c);
				Cut(a,b,c);
			}
			else{
				scanf("%d%d",&a,&b);
				Reversal(a,b);
			}
		}
		cnt=0;
		InOrder(root);
		printf("\n");
	}
	return 0;
}

 

 

HDU 3436

 

这道题是离散化数据来进行Splay。  非常感谢这位大牛的代码。

很有借鉴的意义。。  区间的离散， 把需要的点单独存，end跟beg都为1，不然就保存的时候， 1到a1-1就保存成 beg=1, end=a1-1这样类似的。。

理解后就感觉不难了。   

原博客:http://blog.sina.com.cn/s/blog_6c7729450100uvye.html

代码:

//
//  3436.cpp
//  ACM_HDU
//
//  Created by ipqhjjybj on 13-9-7.
//  Copyright (c) 2013年 ipqhjjybj. All rights reserved.
//  先看完别人代码的。。
//
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
using namespace std;
const int MAXN = 222222;
int n,q,k,N;
char re[MAXN];
int req[MAXN],a[MAXN],na,beg[MAXN],end[MAXN];
int num[MAXN],pre[MAXN],ch[MAXN][2],root;
void flow(int x){
    num[x]=num[ch[x][0]]+num[ch[x][1]]+end[x]-beg[x]+1;
}
inline void rotate(int n)
{
    int t=pre[n];
    bool isr=(ch[t][1]==n);
    ch[t][isr]=ch[n][!isr],pre[ch[n][!isr]]=t;
    pre[n]=pre[t],ch[pre[t]][ch[pre[t]][1]==t]=n;
    pre[t]=n,ch[n][!isr]=t;
    flow(t);
}
inline void splay(int n,int goal)
{
    int f,ff;
    while(pre[n]!=goal)
    {
        f=pre[n],ff=pre[f];

        if(goal==ff)
            rotate(n);
        else if((ch[ff][1]==f)==(ch[f][1]==n))
            rotate(f),rotate(n);
        else
            rotate(n),rotate(n);
    }
    flow(n);
    if(!goal)
        root=n;
}

void Top(int t){
    splay(t,0);
    int x=ch[t][1];
    while(ch[x][0])
        x=ch[x][0];
    splay(x,t);
    ch[x][0]=ch[t][0],pre[ch[t][0]]=x;
    ch[t][0]=0;
    //因为算法的特殊性.root不能变为t;
    flow(t);
}

int Rank(int n){
    int t=root;
    while(1){
        if(num[ch[t][0]]<n&&num[ch[t][0]]+end[t]-beg[t]+1>=n)
            break;
        if(n<=num[ch[t][0]])
            t=ch[t][0];
        else {
            n-=num[ch[t][0]]+end[t]-beg[t]+1,t=ch[t][1];
        }
    }
    int ans=beg[t]+n-1-num[ch[t][0]];
    splay(t,0);
    return ans;
}
int Query(int x){
    splay(x,0);
    return num[ch[x][0]]+1;
}

int Bisearch(int n)
{
    int l=1,r=N,mid;

    while(l<=r)
    {
        mid=(l+r)>>1;
        if(beg[mid]>n)
            r=mid-1;
        else if(end[mid]<n)
            l=mid+1;
        else
            return mid;
    }
    return l;
}
int main(){
    int t,tt=0,val,i,j;
    char str[10];
    scanf("%d",&t);
    while(t--){
        scanf("%d %d",&n,&q);
        k=0;
        na=0;
        for(i=0;i<q;i++){
            scanf("%s %d",str,&val);
            re[k]=str[0],req[k++]=val;
            if(str[0]=='T'){
                a[na++]=val;
            }
        }
        sort(a,a+na);
        for(j=i=1;i<na;i++)
            if(a[i]!=a[i-1])
                a[j++]=a[i];
        na=j;j=1;
        if(a[0]>1){
            beg[j]=1,end[j++]=a[0]-1;
        }
        for(i=0;i<na;i++){
            end[j]=beg[j]=a[i],j++;
            if(i<na-1){
                if(a[i]+1<a[i+1]){
                    beg[j]=a[i]+1,end[j]=a[i+1]-1;
                    j++;
                }
            }else{
               beg[j]=a[i]+1,end[j]=n,j++;
            }
        }
        beg[j]=end[j]=n+1;j++;
        memset(ch,0,sizeof(int)*(j+2)*2);
        num[0]=0,N=j-1;
        for(i=1;i<=N;i++){
            num[i]=num[i-1]+end[i]-beg[i]+1;
            pre[i-1]=i,ch[i][0]=i-1;
        }pre[N]=0;
        root=N;
        printf("Case %d:\n",++tt);
        for(i=0;i<k;i++){
            if(re[i]=='T'){
                Top(Bisearch(req[i]));
            }else if(re[i]=='R'){
                printf("%d\n",Rank(req[i]));
            }else{
                j=Bisearch(req[i]);
                printf("%d\n",Query(j)+req[i]-beg[j]);
            }
        }
    }
    return 0;
}