【POJ 2104】 K-th Number 主席树模板题

#include<cstdio>
#include<cstring>
#include<algorithm>
#define for1(i,a,n) for(int i=(a);i<=(n);i++)
#define for2(i,a,n) for(int i=(a);i<(n);i++)
#define for3(i,a,n) for(int i=(a);i>=(n);i++)
#define for4(i,a,n) for(int i=(a);i>(n);i++)
#define CC(i,a) memset(i,a,sizeof(i))
#define MID (l+r)>>1
#define read(x) x=getint()
using namespace std;
const int N=100010;
inline int max(const int &a,const int &b){return a>b?a:b;}
inline int min(const int &a,const int &b){return a<b?a:b;}
inline void swapp(int &a,int &b){int c=a;a=b;b=c;}
inline int getint(){char c;int ret=0,k=1;for(c=getchar();c<‘0‘||c>‘9‘;c=getchar())if(c==‘-‘)k=-1;for(;c>=‘0‘&&c<=‘9‘;c=getchar())ret=ret*10+c-‘0‘;return ret*k;}
int key,a[N],id[N],b[N],n,m,root[N],cnt=0;
struct node{int l,r,s;}T[N*50];
inline bool cmp(int x,int y){return a[x]<a[y];}
inline void update(int l,int r,int &pos){
	T[++cnt]=T[pos]; pos=cnt; T[pos].s++;
	if (l==r) return;
	int m=MID;
	if (key<=m) update(l,m,T[pos].l); else update(m+1,r,T[pos].r);
}
inline int query(int l,int r,int x,int y,int k){
	if (l==r) return l;
	int m=MID,s=T[T[y].l].s-T[T[x].l].s;
	if (k<=s) return query(l,m,T[x].l,T[y].l,k); else return query(m+1,r,T[x].r,T[y].r,k-s);
}
int main(){
	read(n); read(m);
	for1(i,1,n) read(a[i]),id[i]=i;
	sort(id+1,id+n+1,cmp);
	for1(i,1,n) b[id[i]]=i;
	for1(i,1,n){
		root[i]=root[i-1]; key=b[i];
		update(1,n,root[i]);
	}int x,y;
	while (m--){
		read(x); read(y); read(key);
		printf("%d\n",a[id[query(1,n,root[x-1],root[y],key)]]);
	}return 0;
}