主席树模板：【洛谷 P3384】

主席树模板：【洛谷 P3384】

题目大意：

给你一堆$[l,r]$，问区间内的第 k 小数

思路：

详见这篇：戳这里

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<vector>
#define r register
#define rep(i,x,y) for(r ll i=x;i<=y;++i)
#define per(i,x,y) for(r ll i=x;i>=y;--i)
using namespace std;
typedef long long ll;
const ll V=2*1e5+10;
ll tree[V<<5],a[V<<5],b[V<<5];
ll lc[V<<5],rc[V<<5],x,y,k;
ll n,m,top,cnt,f[V<<5],ans,p;
void build(ll &now,ll x,ll y)
{
	now=++cnt;
	ll mid=x+y>>1;
	if(x==y) return ;
	build(lc[now],x,mid);
	build(rc[now],mid+1,y);
}
ll add(ll now,ll x,ll y)
{
	ll nxt=++cnt;
	lc[nxt]=lc[now],rc[nxt]=rc[now];
	tree[nxt]=tree[now]+1;
	if(x==y) return nxt;
	ll mid=x+y>>1;
	if(p<=mid) lc[nxt]=add(lc[nxt],x,mid);
	else rc[nxt]=add(rc[nxt],mid+1,y);
	return nxt;
}
ll ask(ll x,ll y,ll le,ll ri,ll k)
{
	ll ans,val=tree[lc[y]]-tree[lc[x]];
	if(le==ri) return le;
	ll mid=le+ri>>1;
	if(k<=val) ans=ask(lc[x],lc[y],le,mid,k);
	else ans=ask(rc[x],rc[y],mid+1,ri,k-val);
	return ans;
}
int main()
{
	scanf("%lld%lld",&n,&m);
	rep(i,1,n) 
	{
		scanf("%lld",&a[i]);
		b[i]=a[i];
	}
	sort(b+1,b+n+1);
	top=unique(b+1,b+n+1)-b-1;
	build(f[0],1,top);
	rep(i,1,n)
	{
		p=lower_bound(b+1,b+top+1,a[i])-b;
		f[i]=add(f[i-1],1,top);
	}
	rep(i,1,m)
	{
		scanf("%lld%lld%lld",&x,&y,&k);
		ans=ask(f[x-1],f[y],1,top,k);
		printf("%lld\n",b[ans]);
	}
	return 0;
}