主席树模板

#include <bits/stdc++.h>

using namespace std;

#define MP make_pair
#define PB push_back
typedef long long LL;
typedef pair<int,int> PII;
const double eps=1e-8;
const double pi=acos(-1.0);
const int K=2e6+7;
const int mod=1e9+7;

int tot,ls[K],rs[K],rt[K],sum[K];
int a[K],b[K];
//sum[o]记录的是该节点区间内出现的数的个数
//区间指的是将数离散化后的区间
void build(int &o,int l,int r)
{
    o=++tot,sum[o]=0;
    int mid=l+r>>1;
    if(l!=r)
        build(ls[o],l,mid),build(rs[o],mid+1,r);
}
void update(int &o,int l,int r,int last,int x)
{
    o=++tot,sum[o]=sum[last]+1;
    ls[o]=ls[last],rs[o]=rs[last];
    if(l==r) return ;
    int mid=l+r>>1;
    if(x<=mid) update(ls[o],l,mid,ls[last],x);
    else update(rs[o],mid+1,r,rs[last],x);
}
int query(int lo,int ro,int l,int r,int k)
{
    if(l==r) return b[l];
    int cnt=sum[ls[ro]]-sum[ls[lo]],mid=l+r>>1;
    if(k<=cnt) return query(ls[lo],ls[ro],l,mid,k);
    return query(rs[lo],rs[ro],mid+1,r,k-cnt);
}
int main(void)
{
    int T;cin>>T;
    while(T--)
    {
        tot=0;
        int n,m,sz;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
            scanf("%d",a+i),b[i]=a[i];
        sort(b+1,b+1+n);
        sz=unique(b+1,b+1+n)-b-1;
        for(int i=1;i<=n;i++)
            a[i]=lower_bound(b+1,b+1+sz,a[i])-b;
        build(rt[0],1,sz);
        for(int i=1;i<=n;i++)
            update(rt[i],1,sz,rt[i-1],a[i]);
        int l,r,k;
        while(m--)
            scanf("%d%d%d",&l,&r,&k),printf("%d
",query(rt[l-1],rt[r],1,sz,k));

    }
    return 0;
}