hdu 5493 Queue treap实现将元素快速插入到第i个位置
input
T 1<=T<=1000
n 1<=n<=100000
h1 k1
h2 k2
... ...
hn kn
1<=hi<=1e9 0<=ki<=n-1
sum(n)<=1e6
hi!=hj(i!=j)
output
hi指第i个人的身高，ki指这个人前面或者后面比他高的人数
Case #cas: 输出可能的最小序列，没有输出impossible
做法：将所有人按身高排序，从高到低插入数组中，则插入到第i个人时，数组里所有人都比他高，用treap实现，每个人有两个位置可以插入，每次插入到小的位置
1 #include <cstdio> 2 #include <queue> 3 #include <cstring> 4 #include <iostream> 5 #include <cstdlib> 6 #include <algorithm> 7 #include <vector> 8 #include <map> 9 #include <set> 10 #include <ctime> 11 #include <cmath> 12 #include <cctype> 13 #define MAX 100000 14 #define LL long long 15 int cas=1,T,first,a[MAX+10][2]; 16 struct node 17 { 18 node*ch[2]; 19 int r,v,sz; 20 void maintain() { sz=(ch[0]?ch[0]->sz:0)+(ch[1]?ch[1]->sz:0)+1; } 21 }; 22 void rotate(node* &o,int d) 23 { 24 node*k=o->ch[d^1];o->ch[d^1]=k->ch[d];k->ch[d]=o; 25 o->maintain();k->maintain();o=k; 26 } 27 void insert(node* &o,int pos,int &x) 28 { 29 if(o==NULL) 30 { 31 o=new node(); 32 o->ch[0]=o->ch[1]=NULL; 33 o->v=x;o->r=rand(); 34 o->sz=1; 35 return; 36 } 37 o->sz++; //易错，旋转时可能没旋转到该结点，导致没更新，所以在插入后要＋＋ 38 int d; 39 if(pos <= (o->ch[0]?o->ch[0]->sz:0)) d=0; 40 else { d=1;pos -= (o->ch[0]?o->ch[0]->sz:0)+1; } 41 insert(o->ch[d],pos,x); 42 if(o->ch[d]->r > o->r) rotate(o,d^1); 43 } 44 void read(node*u) 45 { 46 if(u==NULL) return; 47 read(u->ch[0]); 48 printf(" %d",u->v); 49 read(u->ch[1]); 50 delete u; 51 } 52 int cmp(const void*a,const void*b) { return *(int*)b-*(int*)a; } 53 int main() 54 { 55 //freopen("/home/user/桌面/in","r",stdin); 56 scanf("%d",&T); 57 int n; 58 while(T--) 59 { 60 scanf("%d",&n); 61 for(int i=1;i<=n;i++) scanf("%d%d",&a[i][0],&a[i][1]); 62 qsort(a+1,n,sizeof(a[0]),cmp); 63 int i; 64 node *root=NULL; 65 for(i=1;i<=n;i++) 66 { 67 if(a[i][1]>=i) break; 68 int pos=std::min(a[i][1],i-1-a[i][1]); 69 insert(root,pos,a[i][0]); 70 } 71 printf("Case #%d:",cas++); 72 if(i<=n) { puts(" impossible");continue; } 73 read(root); 74 printf("\n"); 75 } 76 //printf("time=%.3lf",(double)clock()/CLOCKS_PER_SEC); 77 return 0; 78 }
