P1533 可怜的狗狗

*原题链接*

题意：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;
}