【C++STL/红黑树】POJ 3481 DoubleQueue

POJ 3481 Double Queue

描述：

新成立的BIG-Bank在不切雷斯特开了一间新办公室,使用了由IBM罗马尼亚的现代计算机办公环境,运用了现代信息技术.一般来说,银行的每个顾客都有一个识别码K,并且每一个来银行的顾客都会被给予一个优先级P.银行主管的一个大胆想法震惊了公司的软件工程师.他希望有时候能打破传统,让服务窗口给优先级最低的顾客而不是优先级最高的服务.因此,管理系统会收到以下几种请求.

0	系统停止运行
1 K P	将编号为K的顾客以P的优先级加入等待名单
2	为优先级最高的顾客服务并且将其从等待名单中移除
3	为优先级最低的顾客服务并且将其从等待名单中移除

你的任务是帮助软件工程师来编写这个顾客管理系统.

输入：

输入的每一行有一个请求；只有最后一行会要求停止运行（请求0）.你需要保证当添加一个新顾客到名单中（请求1）时,名单中没有编码相同或优先级相同的顾客..

输出：

对于请求2和请求3,程序需要在不同行标准输出客户编号K,如果此时名单为空,那么输出0.

Sample Input

2
1 20 14
1 30 3
2
1 10 99
3
2
2
0

Sample Output

0
20
30
10
0

这道题很显然是平衡树问题，对于平衡数，C++的STL库其实有一个自带的封装红黑树，就是set，这道题如果用STL库的set就非常简单了。
注意几点，这道题需要记录客户数据，所以可以新开一个数组来记录（因为set中的元素就是键值也就是优先级，所以要单独记录顾客编号）。

这里面用到的主要是set里的insert，end，begin和erase操作，简单讲一下
insert(p)向容器中插入元素p

begin()返回第一个元素的迭代器地址

end()返回最后一个元素的迭代器地址

erase(p)从容器中删除元素p，这里要注意，set中元素的值是唯一的。

#include<set>
#include<iostream>
#define MAXN 10000010
using namespace std;
int a[MAXN],c[1000010],k,p,op;
set<int>li(a,a+MAXN);
set<int>::iterator it;
int main(){
    li.clear();
    while(op=(getchar()-'0')){
        if(op==1){
            scanf("%d %d",&k,&p);
            c[p]=k;
            li.insert(p);
        }
        if(op==2){
            if(li.empty()){
                printf("0
");
                continue;
            }
            it=li.end();
            it--;
            printf("%d
",c[*it]);
            li.erase(*it);
        }
        if(op==3){
            if(li.empty()){
                printf("0
");
                continue;
            }
            it=li.begin();
            printf("%d
",c[*it]);
            li.erase(*it);
        }
    }
}