易错点:
- 查询时若k>lsum则下一步应当query(tr[p].rs,tr[q].rs,mid+1,r,k-lsum);这是由可持久化结构的迭代性得出的.
#include<cstdio>
#include<iostream>
using namespace std;
const int MAXN=1e7,INF=2e9;
struct Node{
int ls,rs,cnt;
}tr[MAXN];
int root[MAXN],nodeCnt=0;
void insert(int p,int &q,int l,int r,int val){
if(!q)q=++nodeCnt;
if(l==r){
tr[q].cnt=tr[p].cnt+1;
return;
}
int mid=(l+r)>>1;
if(val<=mid){
tr[q].rs=tr[p].rs;
insert(tr[p].ls,tr[q].ls,l,mid,val);
}else{
tr[q].ls=tr[p].ls;
insert(tr[p].rs,tr[q].rs,mid+1,r,val);
}
tr[q].cnt=tr[tr[q].ls].cnt+tr[tr[q].rs].cnt;
}
int query(int p,int q,int l,int r,int k){
if(l==r)return l;
int mid=(l+r)>>1,lsum=tr[tr[q].ls].cnt-tr[tr[p].ls].cnt;
if(k<=lsum){
return query(tr[p].ls,tr[q].ls,l,mid,k);
}else return query(tr[p].rs,tr[q].rs,mid+1,r,k-lsum);//
}
int main(){
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
int nowValue;
scanf("%d",&nowValue);
insert(root[i-1],root[i],-INF,INF,nowValue);
}
for(int i=1;i<=m;i++){
int l,r,k;
scanf("%d%d%d",&l,&r,&k);
int ans=query(root[l-1],root[r],-INF,INF,k);
printf("%d\n",ans);
}
return 0;
}