区间第k小（主席树模板）

Posted 2021-11-23 thusloop

#include<bits/stdc++.h>
#define int long long
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const int inf=2e18+100;
const int maxn=2e5+100;
int a[maxn],b[maxn];
map<int,int>mp;
struct node
{
	int l,r,sum;
}t[maxn*32];
int root[maxn];
int cnt;
void update(int l,int r,int &x,int y,int num)
{
	t[++cnt]=t[y];t[cnt].sum++;x=cnt;
	if(l==r)return ;
	int mid=(l+r)>>1;
	if(num<=mid)update(l,mid,t[x].l,t[y].l,num);
	else update(mid+1,r,t[x].r,t[y].r,num);
} 
int query(int l,int r,int x,int y,int k)
{
	if(l==r)return l;
	int sum=t[t[y].l].sum-t[t[x].l].sum;
	int mid=(l+r)>>1;
	if(k<=sum) return query(l,mid,t[x].l,t[y].l,k);
	else return query(mid+1,r,t[x].r,t[y].r,k-sum);
}
signed main()
{
	IOS
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		cin>>b[i];
		mp[b[i]]=1;
	}
	int k=0;
	for(auto it:mp)
	{
		mp[it.first]=++k;
		a[k]=it.first;
	}
	for(int i=1;i<=n;i++)
	{
		update(1,n,root[i],root[i-1],mp[b[i]]);
	}
	while(m--)
	{
		int l,r,k;
		cin>>l>>r>>k;
		cout<<a[query(1,n,root[l-1],root[r],k)]<<"\\n";
	}
}