poj 2104 划分树

思路：裸的划分树

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
#define Maxn 100010
#define lson(x) x<<1
#define rson(x) x<<1|1
using namespace std;
int n,m,val[][Maxn],toLeft[][Maxn],sorted[Maxn];
struct Tree{
    int l,r;
    int mid(){
        return (l+r)>>;
    }
}tree[Maxn*];
void BuildTree(int l,int r,int dep,int po)
{
    tree[po].l=l,tree[po].r=r;
    if(l==r)
        return ;
    int mid=tree[po].mid();
    int lpos=l,rpos=mid+;
    int same=mid-l+,i;
    for(i=l;i<=r;i++){
        if(val[dep][i]<sorted[mid])
            same--;
    }
    for(i=l;i<=r;i++){
        if(i==l) toLeft[dep][i]=;
        else toLeft[dep][i]=toLeft[dep][i-];
        if(val[dep][i]<sorted[mid]){
            val[dep+][lpos++]=val[dep][i],toLeft[dep][i]++;
        }
        else
        if(val[dep][i]>sorted[mid]){
            val[dep+][rpos++]=val[dep][i];
        }
        else
        if(same){
            val[dep+][lpos++]=val[dep][i],toLeft[dep][i]++,same--;
        }
        else
            val[dep+][rpos++]=val[dep][i];
    }
    BuildTree(l,mid,dep+,lson(po));
    BuildTree(mid+,r,dep+,rson(po));
}
int query(int l,int r,int k,int dep,int po)
{
    //cout<<tree[po].l<<" "<<tree[po].r<<endl;
    if(l==r)
        return val[dep][l];
    int ls,lns;
    if(l==tree[po].l){
        ls=toLeft[dep][r];
        lns=;
    }else{
        ls=toLeft[dep][r]-toLeft[dep][l-];
        lns=toLeft[dep][l-];
    }
    int mid=tree[po].mid();
    if(ls>=k){
        int newl=tree[po].l+lns;
        int newr=tree[po].l+lns+ls-;
        return query(newl,newr,k,dep+,lson(po));
    }
    else{
        int rns=l-tree[po].l-lns;
        int rs=r-l+-ls;
        int newl=mid++rns;
        int newr=mid+rns+rs;
        return query(newl,newr,k-ls,dep+,rson(po));
    }
}
int main()
{
    int i,j,a,b,k;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(i=;i<=n;i++){
            scanf("%d",&val[][i]);
            sorted[i]=val[][i];
        }
        sort(sorted+,sorted++n);
        BuildTree(,n,,);
        for(i=;i<m;i++){
            scanf("%d%d%d",&a,&b,&k);
            printf("%d\n",query(a,b,k,,));
        }
    }
    return ;
}