阿生的数据结构

阿生在HBU教授数据结构课程。他给可怜的孩子们留了一道毒瘤作业，要求同学给他编写一个新的数据结构，该结构能够在数组的某个区间内快速返回k阶统计量。

也就是说，给定一个由不同整数组成的数组a[1⋯n]，给出一系列问题Q(i，j，k)需要给出区间[i,j]中的第k小的数字。

例如，考虑数组a=(1，5，2，6，3，7，4)。设问题为Q（2，5，3）。区间a[2⋯5]是(5，2，6，3)。问题的答案是5。

输入格式:

输入文件的第一行包含n数组的大小，m要回答的问题数(1<=n<=100000，1<=m<=5000)。

第二行包含n个不同的整数，其绝对值不超过109.

以下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小，显然是主席树，板子

#include<bits/stdc++.h>
using namespace std;

#define x first
#define y second
#define endl '\n'
#define rep(i,a,n) for (int i = a; i < n; i ++ )
#define repn(i,a,n) for (int i = a; i <= n; i ++ )
#define pb push_back
#define IOS ios::sync_with_stdio(false); cin.tie(0);cout.tie(0);
typedef long long ll;
#define all(x) (x).begin(),(x).end()
typedef pair<int,int> PII;

ll gcd(ll a,ll b) { return b ? gcd(b,a % b) : a; }
const int mod = 1e9+7;
const int N = 100010, M = 10010;

int n, m;
int a[N];
vector<int> nums;

struct Node
{
    int l, r;
    int cnt;
}tr[N * 4 + N * 17];

int root[N], idx;

int find(int x)
{
    return lower_bound(nums.begin(), nums.end(), x) - nums.begin();
}

int build(int l, int r)
{
    int p = ++ idx;
    if (l == r) return p;
    int mid = l + r >> 1;
    tr[p].l = build(l, mid), tr[p].r = build(mid + 1, r);
    return p;
}

int insert(int p, int l, int r, int x)
{
    int q = ++ idx;
    tr[q] = tr[p];
    if (l == r)
    {
        tr[q].cnt ++ ;
        return q;
    }
    int mid = l + r >> 1;
    if (x <= mid) tr[q].l = insert(tr[p].l, l, mid, x);
    else tr[q].r = insert(tr[p].r, mid + 1, r, x);
    tr[q].cnt = tr[tr[q].l].cnt + tr[tr[q].r].cnt;
    return q;
}

int query(int q, int p, int l, int r, int k)
{
    if (l == r) return r;
    int cnt = tr[tr[q].l].cnt - tr[tr[p].l].cnt;
    int mid = l + r >> 1;
    if (k <= cnt) return query(tr[q].l, tr[p].l, l, mid, k);
    else return query(tr[q].r, tr[p].r, mid + 1, r, k - cnt);
}

int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++ )
    {
        scanf("%d", &a[i]);
        nums.push_back(a[i]);
    }

    sort(nums.begin(), nums.end());
    nums.erase(unique(nums.begin(), nums.end()), nums.end());

    root[0] = build(0, nums.size() - 1);

    for (int i = 1; i <= n; i ++ )
        root[i] = insert(root[i - 1], 0, nums.size() - 1, find(a[i]));

    while (m -- )
    {
        int l, r, k;
        scanf("%d%d%d", &l, &r, &k);
        printf("%d\n", nums[query(root[r], root[l - 1], 0, nums.size() - 1, k)]);
    }

    return 0;
}