poj 1442

两种操作， add：是向序列中加入一个数 ， get是问第k小数是谁 ， 可以用优先队列  ， 也可以用treap ， 还可以用线段树 。

线段树的做法为 ， 先离散化 ， 然后每次都往线段树中加入数 ， 线段树相应的的点+1 ， 然后可以二分找 。 思想很明显， 代码实现注意细节就行 ， 不过我没去写了。这里只给出2种解法。

1：优先队列：两个优先队列， 一个最小优先 ， 一个最大优先， 最大优先保存前i-1小的数 ， 最小优先里面的数都比最大优先的大 ，那么每次询问第i小的数时 ， 最小优先队列的第一个就是答案。

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<queue>
 4 #include<cstring>
 5 #include<algorithm>
 6 
 7 const int N = 33333 ;
 8 
 9 using namespace std ;
10 
11 priority_queue<int , vector<int> , greater<int> >Min_Q ;
12 priority_queue<int , vector<int> , less<int> >Max_Q ;
13 
14 int val[N] , n , m ;
15 
16 int main()
17 {
18     while( scanf("%d%d" , &n , &m)==2 )
19     {
20            for( int i=1;i<=n;i++ )    scanf("%d" , &val[i]);
21            
22            while( !Min_Q.empty() ) Min_Q.pop() ;
23            while( !Max_Q.empty() ) Max_Q.pop() ;
24            
25            int p = 0 , rank = 1 , v ;
26            
27            for( int i=1;i<=m;i++ ){
28                 
29                 
30                                  scanf("%d" , &v) ;
31                                 
32                                  while( v>p ){
33                                            int num = val[++p] ;
34                                            
35                                            if( Max_Q.size()<rank-1 ) Max_Q.push( num ) ;
36                                            
37                                            else if( Max_Q.empty() || num>=Max_Q.top() ){
38                                                                Min_Q.push( num ) ;    
39                                            } 
40                                            else{
41                                                                Min_Q.push( Max_Q.top() );
42                                                                Max_Q.pop() ;
43                                                                Max_Q.push( num ) ;     
44                                            } 
45                                  }  
46                                  
47                                  printf("%d\n" , Min_Q.top()) ;
48                                  Max_Q.push( Min_Q.top() ) ;
49                                  Min_Q.pop() ;
50                                  rank++ ;
51            }  
52     }
53 }

2：treap， 模板题。

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<string>
 5 #include<algorithm>
 6 
 7 const int N = 33333 ;
 8 
 9 using namespace std ;
10 
11 int val[N] , n , m , ans ;
12 
13 struct node{
14        node * ch[2] ;
15        int s , v , r ;
16        node(){
17               ch[0] = ch[1] = NULL ;
18               s = 1 ;       
19        }       
20        int cmp( int x )const{
21               if( x==v ) return -1 ;
22               return x<v?0:1 ;      
23        }
24        void maintain(){
25               s = 1 ;
26               if( ch[0]!=NULL ) s += ch[0]->s ;
27               if( ch[1]!=NULL ) s += ch[1]->s ;              
28        }
29 };
30 
31 void rotate( node * &o , int d ){
32      node * k = o->ch[d^1] ;
33      o->ch[d^1] = k->ch[d] ;
34      k->ch[d] = o ;
35      o->maintain() ; k->maintain() ;
36      o = k ;     
37 }
38 
39 bool find_x( node * o , int x ){
40      while( o!=NULL ){
41                      int d = o->cmp(x) ;
42                      if( d==-1 ) return true ;
43                      else o = o->ch[d] ;       
44      }
45      return false ;     
46 }
47 
48 void insert( node * &o , int x ){
49      if( o==NULL ){
50                  o = new node() ;
51                  o->v = x ; o->r = rand() ;
52                  return ;    
53      }     
54      
55      int d = o->cmp(x) ;
56      
57      if( d==-1 )       d = 1 ;
58      insert( o->ch[d] , x ) ;
59      if( o->ch[d]->r > o->r ) rotate( o , d^1 ) ;
60      o->maintain() ;
61 }
62 
63 bool findkth( node * o , int k ){
64      while( o!=NULL ){
65             if( o->s<k ) return false ;
66                     int l = o->ch[0]!=NULL ? o->ch[0]->s : 0 ;
67                     
68                     if( l+1==k ){
69                                ans = o->v ;
70                                return true ;    
71                     }       
72                     
73                     int d = l<k ;
74                     o = o->ch[d] ;
75                     k -= d*(l+1) ;
76      } 
77      return false ;
78 }
79 
80 int main()
81 {
82     while( scanf("%d%d" , &n , &m)==2 ){
83            for( int i=1;i<=n;i++ )    scanf("%d" , &val[i]) ;
84            
85            int p = 0 , rank = 1 , v;
86            
87            node * root = NULL ;
88            for( int i=1;i<=m;i++ ){
89                          scanf("%d" , &v);
90                          
91                          while( v>p ) insert( root , val[++p] ) ;//, cout<<p<<" "<<val[p]<<endl;
92                          
93                          if( findkth( root , rank++ )==true )
94                                       printf("%d\n" , ans);   
95            }       
96     }  
97 }

 

转载于:https://www.cnblogs.com/Crying-Smile/p/4115365.html