【pb_ds】bzoj1056 [HAOI2008]排名系统/bzoj1862 [Zjoi2006]GameZ游戏排名系统

 1 #include<cstdio>
 2 #include<cctype>
 3 #include<iostream>
 4 #include<string>
 5 #include<ext/pb_ds/assoc_container.hpp>
 6 #include<ext/pb_ds/tree_policy.hpp>
 7 using namespace std;
 8 using namespace __gnu_cxx;
 9 using namespace __gnu_pbds;
10 int Res,Num;char C,CH[20];
11 inline int G()
12 {
13     Res=0;C=‘*‘; 
14     while(C<‘0‘||C>‘9‘)C=getchar();
15     while(C>=‘0‘&&C<=‘9‘){Res=Res*10+(C-‘0‘);C=getchar();}
16     return Res;
17 }
18 inline void P(int x)
19 {
20     Num=0;if(!x){putchar(‘0‘);puts("");return;}
21     while(x>0)CH[++Num]=x%10,x/=10;
22     while(Num)putchar(CH[Num--]+48);
23     puts("");
24 }
25 int n,v,en;
26 string s;
27 struct Point
28 {string x;int y,z;
29 Point(const string &a,const int &b,const int &c){x=a;y=b;z=c;}
30 Point(){}};
31 bool operator < (const Point &a,const Point &b){return a.y!=b.y ? a.y<b.y : a.z>b.z;}
32 tree<Point,null_type,less<Point>,rb_tree_tag,tree_order_statistics_node_update> T;//在bzoj上请将null_type改成null_mapped_type
33 typedef tree<Point,null_type,less<Point>,rb_tree_tag,tree_order_statistics_node_update>::iterator ITER;
34 map<string,int>ma,ma2;
35 inline bool is_name(const string &x)
36 {
37     int len=x.length();
38     for(int i=0;i<len;i++) if(!isalpha(x[i])) return false;
39     return true;
40 }
41 inline int to_num(const string &x)
42 {
43     int res=0,base=1,len=x.length();
44     for(int i=len-1;i>=0;i--)
45       {
46           res+=(x[i]-‘0‘)*base;
47           base*=10;
48       }
49     return res;
50 }
51 int main()
52 {
53     n=G();
54     for(int i=1;i<=n;i++)
55       {
56           cin>>s;
57           if(s[0]==‘+‘)
58             {
59                 s.erase(0,1);
60                 v=G();
61                 if(ma.find(s)!=ma.end()) T.erase(T.find(Point(s,ma[s],ma2[s])));
62             ma[s]=v;
63             T.insert(Point(s,v,++en));
64             ma2[s]=en;
65             }
66           else
67             {
68                 s.erase(0,1);
69                 if(is_name(s)) P(T.size()-T.order_of_key(Point(s,ma[s],ma2[s])));
70                 else
71                   {
72                       v=to_num(s);
73                       int i=1; ITER it=T.find_by_order(T.size()-v);
74                       for(;;it--,i++)
75                         {
76                             printf("%s",(*it).x.c_str());
77                             if(it==T.begin()||i==10)
78                       {
79                           puts("");
80                           break;
81                       }
82                     putchar(‘ ‘);
83                         }
84                   }
85             }
86       }
87     return 0;
88 }