COGS 723. [SDOI2007] 超级数组

　　　　　　　　　　　　　　　　　　★★☆   输入文件：arr.in   输出文件：arr.out   简单对比
　　　　　　　　　　　　　　　　　　　　　　时间限制：1 s   内存限制：3 MB

　　　　　　　　　　　　　　　　　　　　  Source: SDOI2007 Day2
【问题描述】

一般的数组大家都经常使用，相信很多同学没有见过下面的超级数组。

超级数组存储的是一些正整数，它还支持下面的两个操作

(1)、插入一个元素，命令是 "i key" 。 key 是要插入的数。

(2)、输出第 k 大元素并删除该元素，命令是 "d k"。输出第 k 大元素并删除它。

“第 k 大”是指：现有的数中，如果从小到大排好序，从最小的开始作为第一大算起，

一直数到第 k 个。

现在给出一个开始是空的超级数组，请维护好该数组。

【输入】(arr.in)

第一行 n、m：n<=1 000 000 000 , m<=100 000。表示插入数的范围是 1 至n ,共有m 条命令（包括插入和删除）。

以下 m 行，每行一条命令，如题中描述。每条命令中字母和后面的数字之间一个空格。

保证输入数据是正确的，删除的数一定存在。

【输出】（arr.out）

对于每个删除命令，按删除命令顺序输出删除的数，每个数一行

【样例输入】

100 10

i 57

i 99

i 65

d 3

i 89

d 2

d 2

d 1

i 93

i 29

【样例输出】

99

65

89

57

SBT模板题。。。

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstdlib>
  4 #include<algorithm>
  5 #include<cmath>
  6 #include<cstring>
  7 #include<queue>
  8 #include<vector>
  9 using namespace std;
 10 const int maxn=100001;
 11 int key[maxn],siz[maxn],lc[maxn],rc[maxn];
 12 int root,tot;
 13 int N,M;
 14 char s[5];
 15 void r_rotate(int &rt){
 16     int k=lc[rt];
 17     lc[rt]=rc[k];
 18     rc[k]=rt;
 19     siz[k]=siz[rt];
 20     siz[rt]=siz[lc[rt]]+siz[rc[rt]]+1;
 21     rt=k;
 22 }
 23 void l_rotate(int &rt){
 24     int k=rc[rt];
 25     rc[rt]=lc[k];
 26     lc[k]=rt;
 27     siz[k]=siz[rt];
 28     siz[rt]=siz[lc[rt]]+siz[rc[rt]]+1;
 29     rt=k;
 30 }
 31 void Maintain(int &rt,bool flag){
 32     if(flag==false){
 33         if(siz[lc[lc[rt]]]>siz[rc[rt]]) r_rotate(rt);
 34         else if(siz[rc[lc[rt]]]>siz[rc[rt]]){
 35             l_rotate(lc[rt]);
 36             r_rotate(rt);
 37         }
 38         else return ;
 39     }
 40     else{
 41         if(siz[rc[rc[rt]]]>siz[lc[rt]]) l_rotate(rt);
 42         else if(siz[lc[rc[rt]]]>siz[lc[rt]]){
 43             r_rotate(rc[rt]);
 44             l_rotate(rt);
 45         }
 46         else return ;
 47     }
 48     Maintain(lc[rt],false); Maintain(rc[rt],true);
 49     Maintain(rt,false); Maintain(rt,true);
 50 }
 51 void insert(int &rt,int v){
 52     if(rt==0){
 53         rt=++tot;
 54         key[rt]=v;
 55         siz[rt]=1; lc[rt]=rc[rt]=0;
 56         return ;
 57     }
 58     siz[rt]++;
 59     if(v<=key[rt]) insert(lc[rt],v);
 60     else insert(rc[rt],v);
 61     Maintain(rt,false); Maintain(rt,true);
 62 }
 63 int Delete(int &rt,int v){
 64     int ans;
 65     siz[rt]--;
 66     if(v==key[rt]||(v<key[rt]&&lc[rt]==0)||(v>key[rt]&&rc[rt]==0)){
 67         ans=key[rt];
 68         if(lc[rt]==0||rc[rt]==0) rt=lc[rt]+rc[rt];
 69         else key[rt]=Delete(lc[rt],key[rt]+1);
 70         return ans;
 71     }
 72     if(v<key[rt]) ans=Delete(lc[rt],v);
 73     else ans=Delete(rc[rt],v);
 74     return ans;
 75 }
 76 bool find(int &rt,int v){
 77     if(rt==0) return false;
 78     if(v==key[rt]) return true;
 79     if(v<key[rt]) return find(lc[rt],v);
 80     if(v>key[rt]) return find(rc[rt],v);
 81 }
 82 int select(int &rt,int v){
 83     if(v==siz[lc[rt]]+1) return key[rt];
 84     else if(v<siz[lc[rt]]+1) return select(lc[rt],v);
 85     else return select(rc[rt],v-siz[lc[rt]]-1);
 86 }
 87 int main(){
 88     freopen("arr.in","r",stdin);
 89     freopen("arr.out","w",stdout);
 90     scanf("%d%d",&N,&M);
 91     for(int i=1,tmp;i<=M;i++){
 92         scanf("%s%d",s,&tmp);
 93         if(s[0]=='i') insert(root,tmp);
 94         else{
 95             int ans=select(root,tmp);
 96             printf("%d\n",ans);
 97             Delete(root,ans);
 98         }
 99     }
100     return 0;
101 }

转载于:https://www.cnblogs.com/CXCXCXC/p/5134294.html