函数式线段树

#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn = 100000 + 10;

struct node {
    int l, r, w;
    node(int l=0, int r=0, int w=0): l(l), r(r), w(w) {}
}T[maxn*20];

int n, m, cnt;
int a[maxn], b[maxn], rt[maxn];

void creat(int l, int r, int& k) {
    k = cnt++;
    T[k] = node(0, 0, 0);
    if(l == r) return;
    int m = (l + r) >> 1;
    creat(l, m, T[k].l);
    creat(m+1, r, T[k].r);
}
void update(int i, int x, int l, int r, int& k) {
    k = cnt++;
    T[k] = node(T[i].l, T[i].r, T[i].w+1);
    if(l == r) return;
    int m = (l + r) >> 1;
    if(x <= m) 
        update(T[i].l, x, l, m, T[k].l);
    else 
        update(T[i].r, x, m+1, r, T[k].r);
}
int query(int i, int j, int l, int r, int k) {
    if(l == r) return l;
    int m = (l + r) / 2;
    int t = T[T[j].l].w - T[T[i].l].w;
    if(t >= k) 
        return query(T[i].l, T[j].l, l, m, k);
    else 
        return query(T[i].r, T[j].r, m+1, r, k-t);
}
int l, r, k;
int main() {
    scanf("%d%d", &n, &m);
    for(int i=1; i<=n; i++) {
        scanf("%d", &a[i]);
        b[i] = a[i];
    }sort(b+1, b+n+1);
    int num = unique(b+1, b+n+1) - b - 1;
    creat(1, num, rt[0]);
    for(int i=1; i<=n; i++) {
        a[i] = lower_bound(b+1, b+num+1, a[i]) - b;
        update(rt[i-1], a[i], 1, num, rt[i]);
    }
    while(m--) {
        scanf("%d%d%d", &l, &r, &k);
        printf("%d\n", b[query(rt[l-1], rt[r], 1, num, k)]);
    }
    return 0;
}