POJ 2761(Treap离线求区间第k大数)

网上有好多解法,我照着LRJ的白书敲了一个板子吧,3907ms过的,这题还可以SBT过,主席树也是可以过得。划分树不知道行不行,反正做法挺多。。T_T。



/****************************
* author:crazy_石头
* date:2014/05/08
* time:3907 ms
* algorithm:Treap-求区间第大数
* Pro:POJ 2761
***************************/
#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); } struct Q { int l,r,k,id; bool operator<(const Q& B)const { return l 
               
              
             
            
           
          
         
       
      
      
     
     
    
    
   
   

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值