POJ 2104 K-th Number 不高效的线段树写法

const int Max_N = 100008  ;

int  A[Max_N] ;
vector<int> data[Max_N<<2] ;

void  make_tree(int L , int R , int root){
if(L == R){
data[root].clear() ;
data[root].push_back(A[L]) ;
return  ;
}
int mid = (L + R) >> 1 ;
make_tree(L , mid , root<<1) ;
make_tree(mid+1 , R ,root<<1|1) ;
data[root].resize(R-L+1) ;
merge(data[root<<1].begin() , data[root<<1].end() ,
data[root<<1|1].begin() , data[root<<1|1].end() , data[root].begin()) ;
}

int query(int l , int r  , int x , int L , int R , int root){
if(r < L || l > R)  return 0 ;
if(l <= L && R <= r)
return upper_bound(data[root].begin() , data[root].end() , x) - data[root].begin() ;
int mid = (L + R) >> 1 ;
return query(l , r , x , L , mid , root<<1)
+ query(l , r , x , mid+1 , R , root<<1|1) ;
}

int  main(){
int T , N , i , M , l , r , k  , L , R , mid , ans ;
cin>>N>>M;
for(i = 1 ; i <= N ; i++)
scanf("%d" ,&A[i]) ;
make_tree(1 , N , 1) ;
sort(A+1 , A+1+N) ;
while(M--){
scanf("%d%d%d" ,&l ,&r ,&k) ;
L = 1 ;
R = N ;
while(L <= R){
mid = (L + R) >> 1 ;
if(query(l , r , A[mid] , 1 , N , 1) >= k){
R = mid - 1 ;
ans = A[mid] ;
}
else
L = mid + 1 ;
}
printf("%d\n" ,ans) ;
}
return 0 ;
}