（划分树，区间第k小）K-th Number

划分树——求区间第k小值

题目：K-th Number
描述

您正在为数据结构部门的Macrohard公司工作。在关于密钥插入的上一个任务失败后，您被要求编写一个新的数据结构，该结构能够快速返回数组段中的第k个顺序统计信息。 
也就是说，给定一个不同整数的[1 ... n]数组，你的程序必须以下列形式回答一系列问题Q（i，j，k）：“什么是第k个数字[i ... j]段，如果此段已分类？”
例如，考虑数组a =（1,5,2,6,3,7,4）。让问题是Q（2,5,3）。段a [2 ... 5]是（5,2,6,3）。如果我们对这个段进行排序，我们得到（2,3,5,6），第三个数字是5，因此问题的答案是5。

输入
输入文件的第一行包含n ---数组的大小，m ---要回答的问题数
（1 <= n <= 100 000,1 <= m <= 5 000）。 
第二行包含n个不同的整数，它们的绝对值不超过1e9 ---应该给出答案的数组。 
以下m行包含问题描述，每个描述由三个数字组成：i，j和k（1 <= i <= j <= n，1 <= k <= j - i + 1）并表示问题Q （i，j，k）。

输出
对于每个问题输出它的答案---排序的[i ... j]段中的第k个数字。

样本输入
7 3
1 5 2 6 3 7 4
2 5 3
4 4 1
1 7 3

样本输出
5
6
3

题意大概就是给一个数组，然后给出一个区间和k，求区间内第k小的值。

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

#define MAXN 100010
#define lson l,m,d+1
#define rson m+1,r,d+1
int a[20][MAXN],sorted[MAXN],toLeft[20][MAXN];

void build(int l,int r,int d) {
    if(l==r) return ;
    int m=(l+r)>>1,p1=l-1,p2=m;
    int mv=sorted[m];
    if(l==1) toLeft[d][0]=0;
    int lsame=m-l+1;
    for(int i=l;i<=r;++i) if(a[d][i]<mv) --lsame;//找出左子树可容纳多少中位数
    for(int i=l;i<=r;++i) {
        toLeft[d][i]=toLeft[d][i-1];
        if(a[d][i]<mv) ++toLeft[d][i],a[d+1][++p1]=a[d][i];
        else if(a[d][i]>mv) a[d+1][++p2]=a[d][i];
        else {
            if(lsame) ++toLeft[d][i],a[d+1][++p1]=a[d][i],--lsame;//左子树还可容纳中位数
            else a[d+1][++p2]=a[d][i];
        }
    }
    build(lson);
    build(rson);
}

int query(int k,int x,int y,int l,int r,int d) {
    if(l==r) return sorted[l];
    int l1=toLeft[d][x-1]-toLeft[d][l-1],l2=toLeft[d][y]-toLeft[d][x-1]; //l1=[l,x-1]上划入左子树的数目,l2=[x,y]上划入左子树的数目
    int m=(l+r)>>1;
    int r1=x-l-l1,r2=y-x+1-l2; //r1=[l,x-1]上划入右子树的数目,r2=[x,y]上划入右子树的数目
    if(k<=l2) return query(k,l+l1,l+l1+l2-1,lson);
    else return query(k-l2,m+1+r1,m+r1+r2,rson);
}

int main() {
    ios::sync_with_stdio(false);
    int n,m;
    cin >> n >> m;
    for(int i=1;i<=n;++i) cin >> a[1][i],sorted[i]=a[1][i];
    sort(sorted+1,sorted+1+n);
    build(1,n,1);
    while(m--) {
        int a,b,k;
        cin >> a >> b >> k;
        cout << query(k,a,b,1,n,1) << endl;
    }
    return 0;
}