hdu 4585- HDU 4585 Shaolin Treap + 寻找前驱后继 水

 题意：

每次添加一个数在集合中，并寻找这个数在集合排序后的前驱和后继 

treap模版水题

//treap
//插入、删除、查询数排名、查询第k小、查询前驱与后继、查询树大小
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>
#include <queue>
#include <map>
#include <set>
#include <vector>
using namespace std;
#include <iostream>
using namespace std;
#define MAXN 200010
struct  node
{
    int id,val;
    node() {}
    node(int a,int b)
    {
        id=a,val=b;
    };
    bool operator>(const node &b) const
    {
        return val>b.val;
    }
};
struct data
{
    int l,r,size,rnd,w;
    node v;
};

class tp
{
public:
    int n,size,root;
    node ans;
    tp()
    {
        n=root=0;
        size=0;
    }
    data tr[MAXN];
    void update(int k)//更新结点信息
    {
        tr[k].size=tr[tr[k].l].size+tr[tr[k].r].size+tr[k].w;
    }
    void init()
    {
        n=root=0;
        size=0;
        memset(tr,0,sizeof tr);
    }
    void rturn(int &k)
    {
        int t=tr[k].l;
        tr[k].l=tr[t].r;
        tr[t].r=k;
        tr[t].size=tr[k].size;
        update(k);
        k=t;
    }
    void lturn(int &k)
    {
        int t=tr[k].r;
        tr[k].r=tr[t].l;
        tr[t].l=k;
        tr[t].size=tr[k].size;
        update(k);
        k=t;
    }
    void insert(int &k,node x)
    {
        if(k==0)
        {
            size++;
            k=size;
            tr[k].size=tr[k].w=1;
            tr[k].v=x;
            tr[k].rnd=rand();
            return;
        }
        tr[k].size++;
        /*if(tr[k].v==x)tr[k].w++;//每个结点顺便记录下与该节点相同值的数的个数
        else */if(x>tr[k].v)
        {
            insert(tr[k].r,x);
            if(tr[tr[k].r].rnd<tr[k].rnd)lturn(k);//维护堆性质
        }
        else
        {
            insert(tr[k].l,x);
            if(tr[tr[k].l].rnd<tr[k].rnd)rturn(k);
        }
    }
    void del(int &k,node& x)// 删除x数(若有多个相同的数，因只删除一个)
    {
        if(k==0)return;
        /* if(tr[k].v==x)
         {
             if(tr[k].w>1)
             {
                 tr[k].w--;
                 tr[k].size--;
                 return;//若不止相同值的个数有多个，删去一个
             }
             if(tr[k].l*tr[k].r==0)k=tr[k].l+tr[k].r;//有一个儿子为空
             else if(tr[tr[k].l].rnd<tr[tr[k].r].rnd)
                 rturn(k),del(k,x);
             else lturn(k),del(k,x);
         }
         else*/ if(x>tr[k].v)
            tr[k].size--,del(tr[k].r,x);
        else tr[k].size--,del(tr[k].l,x);
    }


    void query_pro(int k,node& x)//求x的前驱(前驱定义为小于x，且最大的数)
    {
        if(k==0)return;
        if(x>tr[k].v)
        {
            ans=tr[k].v;
            query_pro(tr[k].r,x);
        }
        else query_pro(tr[k].l,x);
    }
    void query_sub(int k,node& x)//求x的后继(后继定义为大于x，且最小的数)
    {
        if(k==0)return;
        if(tr[k].v>x)
        {
            ans=tr[k].v;
            query_sub(tr[k].l,x);
        }
        else query_sub(tr[k].r,x);
    }
};

tp sb;
int main()
{
    int n;
    int a,b;
     while(scanf("%d", &n)!=EOF)
    {
        if (!n)break;
        sb.init();
        sb.insert(sb.root,node(0,-1e9));
        sb.insert(sb.root,node(1,1e9));
        node tmp;
        for(int   i = 1; i <= n; i++)
        {
            scanf("%d%d",&a,&b);
            tmp.id=a,tmp.val=b;
            sb.insert(sb.root,tmp);
            sb.query_pro(sb.root,tmp);
            node up=sb.ans;
            sb.query_sub(sb.root,tmp);
            node dn=sb.ans;
            int dif1=abs(up.val-b);
            int dif2=abs(dn.val-b);
            if (dif1<=dif2)
                printf("%d %d\n",a,up.id);
            else if (dif1>dif2)
                printf("%d %d\n",a,dn.id);
        }
    }


    return 0;
}