划分树 [POJ 2104] K-th Number

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define N 100010

int sorted[N];
int val[20][N];
int num[20][N];

void build(int l,int r,int d)
{
    if(l==r) return;
    int m=(l+r)>>1,same=0,i;
    
    for(i=l;i<=r;i++)
    {
        if(val[d][i]<sorted[m]) same++;
    }
    
    same=m-l+1-same;
    int lpos=l,rpos=m+1;
    
    for(i=l;i<=r;i++)
    {
        if(val[d][i]<sorted[m])
            val[d+1][lpos++]=val[d][i];
        else if(val[d][i]==sorted[m] && same>0)
        {
            val[d+1][lpos++]=val[d][i];
            same--;
        }
        else
            val[d+1][rpos++]=val[d][i];
        num[d][i]=num[d][l-1]+lpos-l;
    }
    build(l,m,d+1);
    build(m+1,r,d+1);
}
int query(int L,int R,int d,int l,int r,int k)
{
    if(l==r) 
        return val[d][l];
    int m=(L+R)>>1;
    
    int cnt=num[d][r]-num[d][l-1];
    if(cnt>=k)
    {
        int newl=L+num[d][l-1]-num[d][L-1];
        int newr=newl+cnt-1;
        return query(L,m,d+1,newl,newr,k);
    }
    else
    {
        int newr=r+num[d][R]-num[d][r];
        int newl=newr-(r-l-cnt);
        return query(m+1,R,d+1,newl,newr,k-cnt);
    }
}
int main()
{
    int m,n,i,j;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(i=1;i<=n;i++)
        {
            scanf("%d",&sorted[i]); 
            val[0][i]=sorted[i];
        }
        sort(sorted+1,sorted+n+1); 
        build(1,n,0);
        while(m--)
        {
            int l,r,k;
            scanf("%d%d%d",&l,&r,&k);
            printf("%d\n",query(1,n,0,l,r,k));
        }
    }
}