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

题目地址

---

易错点：

• 查询时若k>lsum则下一步应当query(tr[p].rs,tr[q].rs,mid+1,r,k-lsum);这是由可持久化结构的迭代性得出的.

---

#include<cstdio>
#include<iostream>
using namespace std;
const int MAXN=1e7,INF=2e9;
struct Node{
	int ls,rs,cnt;
}tr[MAXN];
int root[MAXN],nodeCnt=0;
void insert(int p,int &q,int l,int r,int val){
	if(!q)q=++nodeCnt;
	if(l==r){
		tr[q].cnt=tr[p].cnt+1;
		return;
	}
	int mid=(l+r)>>1;
	if(val<=mid){
		tr[q].rs=tr[p].rs;
		insert(tr[p].ls,tr[q].ls,l,mid,val);
	}else{
		tr[q].ls=tr[p].ls;
		insert(tr[p].rs,tr[q].rs,mid+1,r,val);
	}
	tr[q].cnt=tr[tr[q].ls].cnt+tr[tr[q].rs].cnt;
}
int query(int p,int q,int l,int r,int k){
	if(l==r)return l;
	int mid=(l+r)>>1,lsum=tr[tr[q].ls].cnt-tr[tr[p].ls].cnt;
	if(k<=lsum){
		return query(tr[p].ls,tr[q].ls,l,mid,k);
	}else return query(tr[p].rs,tr[q].rs,mid+1,r,k-lsum);//
}
int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		int nowValue;
		scanf("%d",&nowValue);
		insert(root[i-1],root[i],-INF,INF,nowValue);
	}
	for(int i=1;i<=m;i++){
		int l,r,k;
		scanf("%d%d%d",&l,&r,&k);
		int ans=query(root[l-1],root[r],-INF,INF,k);
		printf("%d\n",ans);
	}
	return 0;
}