[POJ2104] 主席树模板题

第一次写主席树，比较畏惧，害怕又会因为一些小错误调试半天，就拿这玩意当模板题写了，然而并没有想象的困难。

---

抱怨几句

指针的写法要判NULL真的麻烦= =
随便写什么树一眼望过去全是

if(XXXX!=NULL)

---

关于题目

题目连接：POJ2104
题目大意：给你一串数字，每次询问区间[L,R]的第K大

---

分析与思路

感觉没有什么可以说的····
就是要注意建树的时候，后一个节点相对于前一个节点的树实际上只改动了一条链，或者说只有一条链与前一棵树不同。

如果元素要进左子树的话，右子树就会跟上个树这个区间的右子树是完全一样的，因此，可以直接将本树本节点的右子树指针接到上棵树当前节点的右儿子，这样即省时间，又省空间。
from Oyking cnblogs
细节看代码。

---

代码

Problem: 2104       User: Nafario
Memory: 49140K      Time: 2469MS
Language: C++       Result: Accepted


#include<algorithm> 
#include<cstring>
#include<cstdio>
#include<iostream>
using namespace std;

int N , M ;
struct node{
    int siz ;
    node *ls , *rs ;
    node(){
        ls = rs = NULL ;
        siz = 0 ;
    }
    void update(){
        siz = 0 ;
        if(ls) siz += ls->siz ;
        if(rs) siz += rs->siz ;
    }

}w[200005 * 20] , *tw = w , *root[200005] ;

int ori[200005] , ns[200005] , arcns[200005] , so[200005] , siz ;

void read_uniq(){
    for( int i = 1 ; i <= N ; i ++ ){
        scanf( "%d" , &ori[i] ) ; 
        so[i] = ns[i] = ori[i] ;
    }
    sort ( so + 1 , so + N + 1 ) ;
    siz = unique( so + 1 , so + N + 1 ) - so - 1 ;
    for( int i = 1 ; i <= N ; i ++ ){
        ns[i] = lower_bound( so + 1 , so + siz + 1 , ns[i] ) - so  ;
        arcns[ ns[i] ] = ori[i] ;
    }
}

void Insert( node *&las , node *&nd , int lf , int rg , int val ) {
    nd = ++tw ;
    las = ( las == NULL  ?  &w[0] : las ) ;
    *nd = node();
    //printf( "nd = %d  lf = %d rg = %d val = %d\n" ,nd , lf , rg , val ) ;
    int mid = ( lf + rg ) >> 1 ;
    if( lf == rg ){
        *nd = *las ;
        nd->siz ++ ;
        return ;
    } else if( val <= mid ){
        Insert( las->ls , nd->ls , lf , mid , val ) ;
        nd->rs = las->rs ;
    } else {
        Insert( las->rs , nd->rs , mid + 1 , rg , val ) ;
        nd->ls = las->ls ;
    }
    nd->update() ;
}

int Query( node* &Lnd , node* &Rnd , int lf , int rg , int k){
    if( lf == rg )
        return  lf ;
    //printf("%d %d   lf = %d   rg = %d\n  k = %d" , Lnd , Rnd , lf , rg , k) ;
    Lnd = ( Lnd == NULL ? &w[0] : Lnd ) ;
    //Rnd = ( Rnd == NULL ? &w[0] : Rnd ) ;
    int Lt = 0 ;
    if( Rnd->ls ) Lt += Rnd->ls->siz ;
    if( Lnd->ls ) Lt -= Lnd->ls->siz ;
    //printf("Lt = %d\n" , Lt) ;
    int mid = ( lf + rg ) >> 1 ; 
    if( k > Lt )    return Query( Lnd->rs , Rnd->rs , mid + 1 , rg , k - Lt ) ;
    else            return Query( Lnd->ls , Rnd->ls , lf , mid , k ) ;
}

void solve(){
    int L , R , k ;
    w[0] = node() ;
    for( int i = 1 ; i <= N ; i ++ )
        Insert( root[i - 1] , root[i] , 1 , siz , ns[i] );
    for( int i = 1 ; i <= M ; i ++ ){
        scanf( "%d%d%d" , &L , &R , &k ) ;
        printf( "%d\n", arcns[ Query( root[ L - 1 ] , root[R] , 1 , siz , k ) ] );
    }
}

int main(){
    std::ios::sync_with_stdio(false);
    scanf( "%d%d" , &N , &M ) ;
    read_uniq() ;
    solve() ;
    return 0 ;
}