P3834 【模板】可持久化线段树 1（主席树）

主席树模板

#include<iostream>
#include<string.h>
#include<stdio.h>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<set>
#include<stack>
#include<map>
using namespace std;
#define ll long long
#define F(n) for(int i=0;i<n;i++)
#define clr(a,len,num) for(int i=0;i<len;i++) a[i]=num;
const int mx=200010;
int a[mx],b[mx];
int rt[mx],lc[mx<<5],rc[mx<<5],data[mx<<5]; 
int node_sz=0,p,q;
void build(int &t,int l,int r){
	t=++node_sz;
	data[t]=0;
	if(l==r) return; 
	int mid=(l+r)>>1;
	build(lc[t],l,mid);
	build(rc[t],mid+1,r);
}
int modify(int t,int l,int r){
	int tt=++node_sz;
	data[tt]=data[t]+1;
	lc[tt]=lc[t],rc[tt]=rc[t];
	if(l==r) return tt;
	int mid=(l+r)>>1;
	if(mid>=p) lc[tt]=modify(lc[tt],l,mid);
	else rc[tt]=modify(rc[tt],mid+1,r);
	return tt;
}
int query(int u,int v,int l,int r,int k){
	int x=data[lc[v]]-data[lc[u]];
	int mid=(l+r)>>1;
	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(){
	int m,n;
	int u,v,k;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		b[i]=a[i];
	}
	sort(b+1,b+n+1);				//排序
	int q=unique(b+1,b+n+1)-b-1;  	//去重
	build(rt[0],1,q);				//建立零树（可以省略）
	for(int i=1;i<=n;i++){
		p=lower_bound(b+1,b+q+1,a[i])-b;
		rt[i]=modify(rt[i-1],1,q);
	}
	for(int i=1;i<=m;i++){
		cin>>u>>v>>k;
		cout<<b[query(rt[u-1],rt[v],1,q,k)]<<endl;
	}
}