划分树模板

源自：kuangbin的ACM模板（新）

题目链接：POJ 2104

#include <iostream>
#include <cstring>
#include <string>
#include <cstdlib>
#include <cstdio>
#include <map>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn = 100010;
const int MAXN = 100010;
int tree[20][MAXN];   //表示每层每个位置的值
int sorted[MAXN];     //已经排序好的数
int toleft[20][MAXN]; //toleft[p][i] 表示第 p 层从 1 到 i 有多少个数分入左边
void build(int l, int r, int dep)
{
    if (l == r)
        return;
    int mid = (l + r) >> 1;
    int same = mid - l + 1;      //表示等于中间值而且被分入左边的个数
    for (int i = l; i <= r; i++) //注意是 l, 不是 one
        if (tree[dep][i] < sorted[mid])
            same--;
    int lpos = l;
    int rpos = mid + 1;
    for (int i = l; i <= r; i++)
    {
        if (tree[dep][i] < sorted[mid])
            tree[dep + 1][lpos++] = tree[dep][i];
        else if (tree[dep][i] == sorted[mid] && same > 0)
        {
            tree[dep + 1][lpos++] = tree[dep][i];
            same--;
        }
        else
            tree[dep + 1][rpos++] = tree[dep][i];
        toleft[dep][i] = toleft[dep][l - 1] + lpos - l;
    }
    build(l, mid, dep + 1);
    build(mid + 1, r, dep + 1);
}
//查询区间第 k 大的数,[L,R] 是大区间,[l,r] 是要查询的小区间

int query(int L, int R, int l, int r, int dep, int k)
{
    if (l == r)
        return tree[dep][l];
    int mid = (L + R) >> 1;
    int cnt = toleft[dep][r] - toleft[dep][l - 1];
    if (cnt >= k)
    {
        int newl = L + toleft[dep][l - 1] - toleft[dep][L - 1];
        //在要查询的区间左边界的左面的数如果进入左子树，说明这些数不需要考虑了，也仍然在
        //要查询的区间左面
        int newr = newl + cnt - 1;
        //左边界加上区间内数的个数
        return query(L, mid, newl, newr, dep + 1, k);
    }
    else
    {
        int newr = r + toleft[dep][R] - toleft[dep][r];
        //int newr = R-(R-r - (toleft[dep][R] - toleft[dep][r]));
        //toleft[dep][R] - toleft[dep][r]为要查询的有边界右面有几个数进入了左子树，
        //有几个数进入，则右边界往右移几位
        int newl = newr - (r - l + 1 - cnt) + 1;
        //左边界为右边界减掉区间内数的个数，但是区间内的数会有一部分进入左子树，则原来
        //数的数量减掉进入左子树的数的数量即为现在区间内数的个数，注意加一减一的问题。
        return query(mid + 1, R, newl, newr, dep + 1, k - cnt);
    }
}
int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    memset(tree, 0, sizeof(tree));
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", &tree[0][i]);
        sorted[i] = tree[0][i];
    }
    sort(sorted + 1, sorted + n + 1);
    build(1, n, 0);
    int s, t, k;
    while (m--)
    {
        scanf("%d%d%d", &s, &t, &k);
        printf("%d\n", query(1, n, s, t, 0, k));
    }

    return 0;
}