255. 第K小数(主席树)

给定长度为 N 的整数序列 A，下标为 1∼N。

现在要执行 M 次操作，其中第 i 次操作为给出三个整数 li,ri,ki，求 A[li],A[li+1],…,A[ri] (即 A 的下标区间 [li,ri])中第 ki 小的数是多少。

输入格式
第一行包含两个整数 N 和 M。

第二行包含 N 个整数，表示整数序列 A。

接下来 M 行，每行包含三个整数 li,ri,ki，用以描述第 i 次操作。

输出格式
对于每次操作输出一个结果，表示在该次操作中，第 k 小的数的数值。

每个结果占一行。

数据范围
N≤105,M≤104,|A[i]|≤109

输入样例：
7 3
1 5 2 6 3 7 4
2 5 3
4 4 1
1 7 3
输出样例：
5
6
3

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 1e5+10, M = 1e4+10;
int w[N], root[N], idx;
vector<int> pos;
int n, m;
struct Node
{
    int l, r, cnt;
}t[N * 4 + 17 * N];
void build(int &u, int L, int R)
{
    u = ++ idx;
    if(L == R) return;
    int mid = L + R >> 1;
    build(t[u].l, L, mid); build(t[u].r, mid + 1, R);
}
int getp(int x)
{
    return lower_bound(pos.begin(), pos.end(), x) - pos.begin();
}
void insert(int p, int &q, int L, int R, int k)
{
    q = ++ idx; t[q] = t[p]; t[q].cnt ++;
    if(L == R) return;
    int mid = L + R >>1;
    if(k <= mid) insert(t[p].l, t[q].l, L, mid, k);
    if(k > mid) insert(t[p].r, t[q].r, mid + 1, R, k);
}
int query(int p, int q, int k, int L, int R)
{
    if(L == R) return L;
    int mid = L + R >> 1;
    int cnt = t[t[q].l].cnt - t[t[p].l].cnt;
    if(cnt >= k) return query(t[p].l, t[q].l, k, L, mid);
    if(cnt < k) return query(t[p].r, t[q].r, k - cnt, mid + 1, R);
    
}
int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; ++ i)
    {
        scanf("%d", &w[i]);
        pos.push_back(w[i]);
    }
    sort(pos.begin(), pos.end());
    pos.erase(unique(pos.begin(), pos.end()), pos.end());
    build(root[0], 0, pos.size() - 1);
    for(int i = 1; i <= n; ++ i) insert(root[i - 1], root[i], 0, pos.size() - 1, getp(w[i]));
    while(m -- )
    {
        int l, r, k; scanf("%d%d%d", &l, &r, &k);
        printf("%d\n", pos[query(root[l - 1], root[r], k, 0, pos.size() - 1)]);
    }
}

标签：cnt,return,int,mid,pos,li,255,主席,小数
来源： https://blog.csdn.net/m0_46656833/article/details/116563123