*原题链接*
题意:m次询问区间[l,r]中第k大的数,各询问区间互不相交。
这道题的解法可谓是八仙过海,而我看到询问区间[l,r]中第k大的数,直接就想到了权值线段树(本蒟蒻不会更高级的数据结构了),权值线段树维护区间中数的出现次数,询问时比较k与左子树和右子树中数的出现次数,非常板子,值域比较大,需要先离散化。然后如何处理若干次区间询问?加个莫队就可以了。时间复杂度,比较暴力,但足以通过此题。
#include<bits/stdc++.h>
using namespace std;
const int N=3e5+10;
int n,m,a[N],lsh[N],tot,ans[N],pos[N],t;
struct Node{
int l,r,k,id;
}que[N];
bool cmp(Node a,Node b){
if(pos[a.l]==pos[b.l]) return a.r<b.r;
return a.l<b.l;
}
struct node{
int l,r,cnt;
}tr[N*4];
void pushup(int u){
tr[u].cnt=tr[u<<1].cnt+tr[u<<1|1].cnt;
}
void build(int u,int l,int r){
if(l==r) tr[u]={l,r,0};
else{
tr[u].l=l,tr[u].r=r;
int mid=(l+r)>>1;
build(u<<1,l,mid),build(u<<1|1,mid+1,r);
}
}
void modify(int u,int x,int v){
if(tr[u].l==x&&tr[u].r==x) tr[u].cnt+=v;
else{
int mid=(tr[u].l+tr[u].r)>>1;
if(x<=mid) modify(u<<1,x,v);
else modify(u<<1|1,x,v);
pushup(u);
}
}
//询问区间[l,r]中第k大的数
int query(int u,int l,int r,int k){
if(tr[u].l==tr[u].r) return tr[u].l;
int mid=(tr[u].l+tr[u].r)>>1;
if(k<=tr[u<<1].cnt) return query(u<<1,l,r,k);
return query(u<<1|1,l,r,k-tr[u<<1].cnt);
}
int main(){
ios::sync_with_stdio(false);
cin>>n>>m;t=sqrt(n);
for(int i=1;i<=n;i++) pos[i]=(i-1)/t+1;
for(int i=1;i<=n;i++){cin>>a[i];lsh[i]=a[i];}
sort(lsh+1,lsh+1+n),tot=unique(lsh+1,lsh+1+n)-lsh-1;
for(int i=1;i<=n;i++) a[i]=lower_bound(lsh+1,lsh+1+tot,a[i])-lsh;
for(int i=1;i<=m;i++) cin>>que[i].l>>que[i].r>>que[i].k,que[i].id=i;
sort(que+1,que+1+m,cmp);
build(1,1,n);//可以不用建树,个人习惯问题
//莫队
for(int i=1,l=1,r=0;i<=m;i++){
while(r<que[i].r) modify(1,a[++r],1);
while(r>que[i].r) modify(1,a[r--],-1);
while(l<que[i].l) modify(1,a[l++],-1);
while(l>que[i].l) modify(1,a[--l],1);
ans[que[i].id]=lsh[query(1,1,n,que[i].k)];
}
for(int i=1;i<=m;i++) cout<<ans[i]<<endl;
return 0;
}