主席树查询区间第K大模板

#include <cstdio>
#include <algorithm>
#include <iostream>
using namespace std;
struct data
{
    int num , pos;
}a[100010];
int lp[4000010] , rp[4000010] , sum[4000010] , root[100010] , val[100010] , tot , cnt;
bool cmp1(data a , data b)
{
    return a.num < b.num;
}
bool cmp2(data a , data b)
{
    return a.pos < b.pos;
}
void pushup(int x)
{
    sum[x] = sum[lp[x]] + sum[rp[x]];
}
void ins(int x , int &y , int l , int r , int p)
{
    y = ++tot;
    if(l == r)
    {
        sum[y] = sum[x] + 1;
        return;
    }
    int mid = (l + r) >> 1;
    if(p <= mid) rp[y] = rp[x] , ins(lp[x] , lp[y] , l , mid , p);
    else lp[y] = lp[x] , ins(rp[x] , rp[y] , mid + 1 , r , p);
    pushup(y);
}
int query(int x , int y , int l , int r , int p)
{
    if(l == r) return val[l];
    int mid = (l + r) >> 1;
    if(sum[lp[y]] - sum[lp[x]] >= p) return query(lp[x] , lp[y] , l , mid , p);
    else return query(rp[x] , rp[y] , mid + 1 , r , p - sum[lp[y]] + sum[lp[x]]);
}
int main()
{
    int n , m , i , x , y , z;
    scanf("%d%d" , &n , &m);
    for(i = 1 ; i <= n ; i ++ )
        scanf("%d" , &a[i].num) , a[i].pos = i;
    sort(a + 1 , a + n + 1 , cmp1);
    val[0] = 0x80000000;//负无穷大 
    for(i = 1 ; i <= n ; i ++ )
    {
        if(a[i].num > val[cnt]) val[++cnt] = a[i].num;
        a[i].num = cnt;
    }
    sort(a + 1 , a + n + 1 , cmp2);
    for(i = 1 ; i <= n ; i ++ )
        ins(root[i - 1] , root[i] , 1 , cnt , a[i].num);
    for(i = 1 ; i <= m ; i ++ )
    {
        scanf("%d%d%d" , &x , &y , &z);
         printf("%d\n" , query(root[x - 1] , root[y] , 1 , cnt , z));
    }
}