POJ2104——K-th Number （主席树模板）

传送门

主席树模板，不想多说了

具体的神仙讲解

#include<algorithm>
#include<cmath>
#include<iostream>
#include<stdio.h>
#include<cstdio>
#include<cstdlib>
#include<string.h>
#include<cstring>
using namespace std;
#define mid ((l+r)>>1)
inline int read(){
	char ch=getchar();
	int res=0,f=1;
	while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();}
	while(isdigit(ch)) res=(res<<3)+(res<<1)+(ch^48),ch=getchar();
	return res*f;
}
int n,m,q,tot,num[200005],a[200005],sum[4000005],lc[4000005],rc[4000005],root[4000050];
inline void init(){
	sort(a+1,a+n+1);
	m=unique(a+1,a+1+n)-a-1;
	for(int i=1;i<=n;i++)num[i]=lower_bound(a+1,a+1+m,num[i])-a;
}
inline void buildtree(int fa,int &now,int l,int r,int num){
	now=++tot;
	sum[now]=sum[fa]+1,lc[now]=lc[fa],rc[now]=rc[fa];
	if(l==r) return;
	if(num<=mid) buildtree(lc[fa],lc[now],l,mid,num);
	else buildtree(rc[fa],rc[now],mid+1,r,num);
}
inline int query(int r1,int r2,int l,int r,int k){
	if(l==r)return l;
	int num=sum[lc[r1]]-sum[lc[r2]];
	if(num>=k) return query(lc[r1],lc[r2],l,mid,k);
	else return query(rc[r1],rc[r2],mid+1,r,k-num);
}
int main(){
	n=read(),q=read();
	for(int i=1;i<=n;i++) num[i]=a[i]=read();
	init();
	for(int i=1;i<=n;i++)buildtree(root[i-1],root[i],1,m,num[i]);
	while(q--){
		int l=read(),r=read(),k=read();
		cout<<a[query(root[r],root[l-1],1,m,k)]<<'\n';
	}
}