[HAOI2008]排名系统

传送门：https://ac.nowcoder.com/acm/problem/19971

先记录下来，以后再学。神奇操作

 

题解

用到了C++ pb_ds库，真是一个神奇的库啊....

#include <ext/pb_ds/assoc_containe.hpp>
#include <ext/pb_ds/tree_policy.hpp>

using namespace __gnu_pbds;

定义一颗红黑树
tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update>t;

int 关键字类型
null_type无映射(低版本g++为null_mapped_type)
less<int>从小到大排序
rb_tree_tag 红黑树（splay_tree_tag）
tree_order_statistics_node_update结点更新
插入t.insert();
删除t.erase();
Rank:t.order_of_key();
第K值:t.find_by_order();
前驱:t.lower_bound();
后继t.upper_bound();

代码

#include<bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;

const int maxn=5e5+10;

struct node
{
    int val,id;
    bool operator < (const node &rhs) const{
        if(val!=rhs.val) return val>rhs.val;
        return id<rhs.id;
    }
};

tree<node,null_type,less<node>,rb_tree_tag,tree_order_statistics_node_update> T;
map<string,int>id;
string name[maxn];
int val[maxn];
int cnt=0,tot=0;

int main()
{
    int n;
    cin>>n;
    while(n--){
        char ch;
        string s;
        cin>>ch>>s;
        if(ch=='+'){
            if(id[s]) T.erase((node){val[id[s]],id[s]}),tot--;
            cin>>val[++cnt],id[s]=cnt,name[cnt]=s;
            T.insert((node){val[cnt],cnt}),tot++;
        }
        else{
            if(s[0]>='0'&&s[0]<='9'){
                int x=0,len=s.size();
                for(int i=0;i<len;i++) x=x*10+s[i]-'0';
                    for(int i=x-1;i<min(tot,x+9);i++) cout<<name[T.find_by_order(i)->id]<<' ';
                cout<<endl;
            }
            else cout<<T.order_of_key((node){val[id[s]],id[s]})+1<<endl;
        }
    }
    return 0;
}