解題思路:好久沒用主席樹了,練練手。關于學習主席樹,推薦部落格:https://blog.csdn.net/qq_39809664/article/details/79934516
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define N 100008
using namespace std;
struct node
{
int l,r,num;
}s[N*20];
struct node2
{
int id,v;
}a[N];
int rnk[N],rt[N],cnt;
bool cmp(node2 a,node2 b)
{
return a.v<b.v;
}
void insert(int num,int &x,int l,int r)
{
s[cnt++]=s[x];
x=cnt-1;
// cout<<l<<" "<<r<<endl;
s[x].num++;
if(l==r) return;
int mid=(l+r)/2;
if(num<=mid)
{
insert(num,s[x].l,l,mid);
}else
{
insert(num,s[x].r,mid+1,r);
}
}
int query(int u,int v,int w,int l,int r)
{
if(l==r) return l;
int t=s[s[v].l].num-s[s[u].l].num;
int mid=(l+r)/2;
if(t>=w)
{
return query(s[u].l,s[v].l,w,l,mid);
}else
{
return query(s[u].r,s[v].r,w-t,mid+1,r);
}
}
int main()
{
//freopen("t.txt","r",stdin);
int T,n,m,l1,r1,k;
scanf("%d",&T);
rt[0]=0;
s[0].l=s[0].r=s[0].num=0;
while(T--)
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i].v);
a[i].id=i;
}
sort(a+1,a+1+n,cmp);
for(int i=1;i<=n;i++)
{
rnk[a[i].id]=i;
}
cnt=1;
for(int i=1;i<=n;i++)
{
rt[i]=rt[i-1];
insert(rnk[i],rt[i],1,n);
}
while(m--)
{
scanf("%d%d%d",&l1,&r1,&k);
printf("%d\n",a[query(rt[l1-1],rt[r1],k,1,n)].v);
}
}
return 0;
}