POJ 1442-Treap(插入+区间第k大)

最新推荐文章于 2018-07-27 21:18:00 发布

一名码农、

最新推荐文章于 2018-07-27 21:18:00 发布

阅读量973

点赞数

分类专栏： ACM_数据结构

本文链接：https://blog.csdn.net/u012350533/article/details/25332891

版权

ACM_数据结构专栏收录该内容

47 篇文章 0 订阅

订阅专栏

还是套板子直接求。

/****************************
* author:crazy_石头
* date：2014/05/08
* time:313 ms
* algorithm:Treap-求区间第大数
* Pro：POJ 1442
***************************/
#include 
   
   
    
    
#include 
    
    
     
     
#include 
     
     
      
      
#include 
      
      
       
       
#include 
       
       
         #include 
        
          #include 
         
           #include 
          
            #include 
           
             using namespace std; #define INF 1<<29 #define eps 1e-8 #define A system("pause") #define rep(i,h,n) for(int i=(h);i<=(n);i++) #define ms(a,b) memset((a),(b),sizeof(a)) #define lson l , m , rt << 1 #define rson m + 1 , r , rt << 1 | 1 #define LL long long #define mod 100000000 const int maxn=100000+5; const int maxm=100+10; struct Node { Node* ch[2];//定义左右子树； int r,v;//每个结点的随机优先级及value； int s;//结点总数； Node(int v):v(v) { ch[0]=ch[1]=NULL; r=rand(); s=1; } bool operator <(const Node& rhs)const { return r 
            
              s; if(ch[1]!=NULL) s+=ch[1]->s; } }; inline void rotate(Node* &o,int d)//d==0表示左旋，d==1表示右旋； { Node* k=o->ch[d^1]; o->ch[d^1]=k->ch[d]; k->ch[d]=o; o->maintain(); k->maintain(); o=k; } inline void insert(Node* &o,int x)//先是普通插入，if(x_v 
             
               v?0:1);//不要用cmp函数，可能会有相同结点； insert(o->ch[d],x);//插入新结点，比较优先级进行旋转； if(o->ch[d]->r>o->r) rotate(o,d^1); } o->maintain();//维护结点o的信息； } inline void remove(Node* &o,int x) { int d=o->cmp(x); if(d==-1) { Node* u=o; if(o->ch[0]!=NULL && o->ch[1]!=NULL) { int d2=(o->ch[0]->r>o->ch[1]->r?1:0); rotate(o,d2); remove(o->ch[d2],x); } else { if(o->ch[0]==NULL) o=o->ch[1]; else o=o->ch[0]; delete u; } } else remove (o->ch[d],x); if(o!=NULL) o->maintain(); } //查找区间第k大数； inline int kth(Node* o,int k) { if(o==NULL || k<=0 || k>o->s) return 0; int s=(o->ch[0]==NULL? 0:o->ch[0]->s); if(k==s+1) return o->v; else if(k<=s) return kth(o->ch[0],k); else return kth(o->ch[1],k-s-1); } int a[maxn],n,m; int main() { Node* rt=NULL; scanf("%d%d",&m,&n); rep(i,1,m) scanf("%d",&a[i]); int st=1,test; rep(i,1,n) { scanf("%d",&test); while(st<=test) { insert(rt,a[st]); st++; } printf("%d\n",kth(rt,i)); } return 0; }