POJ3481 Double Queue （平衡树）【例题精讲】

（这题目莫名让我想到了穿越火线。。。）

题意：编写一个银行应用程序，支持下列操作：

1，加入一个客户(输入：1,k,q) 其中k是客户的编号（不一定是1-n但保证小于一百万），q是客户优先级（整型数好吧）。

2，询问当前优先度最高的客户的编号(输入：2)，并且将这个客户删除。

3，询问当前优先度最低的客户的编号(输入：3)，并且将这个客户删除。

4，结束操作(输入0)。

 

这道题动态插入询问最大最小显然平衡树，几乎任何一种平衡树都可以做掉。只要记录树的size，优先度为节点的k值，每次find最小或最大即可；

 

下面上代码：

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int MAXN=1000005;
int key[MAXN],s[MAXN],l[MAXN],r[MAXN];
int t=0,tt=0,q,qq,sign[MAXN],n,m,w,f;
void rrotate(int &t){
    int k=l[t];l[t]=r[k];r[k]=t;
    s[k]=s[t];s[t]=s[l[t]]+s[r[t]]+1;t=k;
}
void lrotate(int &t){
    int k=r[t];r[t]=l[k];l[k]=t;
    s[k]=s[t];s[t]=s[l[t]]+s[r[t]]+1;t=k;
}
void maintain(int &t,bool flag){
    if(!flag){
        if(s[l[l[t]]]>s[r[t]]) rrotate(t);
        else if(s[r[l[t]]]>s[r[t]]){
            lrotate(l[t]);rrotate(t);
        } else return ;
    } else{
        if(s[r[r[t]]]>s[l[t]]) lrotate(t);
        else if(s[l[r[t]]]>s[l[t]]){
            rrotate(r[t]);lrotate(t);
        } else return ;
    } maintain(l[t],false);maintain(r[t],true);
    maintain(t,false);maintain(t,true);
}
void insert(int &t,int v){
    if(t==0){
        tt++;t=tt;key[t]=v;
        s[t]=1;l[t]=r[t]=0;
    } else{
        s[t]++;if(v<key[t]) insert(l[t],v);
        else insert(r[t],v);maintain(t,v>=key[t]);
    } return ;
}
int del(int &t,int v){
    int ret=0;s[t]--;
    if((v==key[t])||(v<key[t]&&l[t]==0)||(v>key[t]&&r[t]==0)){
        ret=key[t];if(l[t]==0||r[t]==0) t=l[t]+r[t];
        else key[t]=del(l[t],key[t]+1);
    } else{
        if(v<key[t]) return del(l[t],v);
        else return del(r[t],v);
    } return 0;
}
int select(int &t,int k){
    if(k==s[l[t]]+1) return t;
    if(k<=s[l[t]]) return select(l[t],k);
    else return select(r[t],k-1-s[l[t]]);
}
int main(){
    while(~scanf("%d",&f)&&f){
        if(f==1){
            scanf("%d%d",&q,&qq);
            insert(t,qq);sign[tt]=q;
        } else if(f==2){
            int cc;if(s[t]==0) puts("0");
            else printf("%d\n",sign[cc=select(t,s[t])]),del(t,key[cc]);
        } else{
            int cc;if(s[t]==0) puts("0");
            else printf("%d\n",sign[cc=select(t,1)]),del(t,key[cc]);
        }
    } 
    return 0;
}