主席树模板

主席树模板题
Code

#include<bits/stdc++.h>
#define ri register int
#define N 200005
using namespace std;
int rc[N<<5],lc[N<<5],rt[N<<5],sum[N<<5];
int n,m,p,q,sz,a[N],b[N];
inline int read() {
	int x=0,f=1;
	char ch;
	ch=getchar();
	while((ch!='-')&&(!(ch>='0'&&ch<='9')))
		ch=getchar();
	if(ch=='-') f=-1,ch=getchar();
	while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
	return x*f;
}
void build(int &rt,int l,int r) {
	rt=++sz,sum[rt]=0;
	if(l==r) return ;
	int mid=l+r>>1;
	build(lc[rt],l,mid);
	build(rc[rt],mid+1,r);
}
int update(int o,int l,int r) {
	int oo=++sz;
	lc[oo]=lc[o];rc[oo]=rc[o];sum[oo]=sum[o]+1;
	if(l==r) return oo;
	int mid=l+r>>1;
	if(mid>=p) lc[oo]=update(lc[oo],l,mid);
	else rc[oo]=update(rc[oo],mid+1,r);
	return oo;
}
int query(int u,int v,int l,int r,int k) {
	int mid=l+r>>1,x=sum[lc[v]]-sum[lc[u]];
	if(l==r) return l;
	if(x>=k) return query(lc[u],lc[v],l,mid,k);
	else return query(rc[u],rc[v],mid+1,r,k-x);
}
int main() {
	n=read(),m=read();
	for(ri i=1;i<=n;i++)
		a[i]=read(),b[i]=a[i];
	sort(b+1,b+n+1);
	q=unique(b+1,b+n+1)-b-1;
	build(rt[0],1,q);
	for(ri i=1;i<=n;i++) {
		p=lower_bound(b+1,b+q+1,a[i])-b;
		rt[i]=update(rt[i-1],1,q);
	}
	while(m--) {
		int l=read(),r=read(),k=read();
		printf("%d\n",b[query(rt[l-1],rt[r],1,q,k)]);
	}
}