AcWing 255. 第K小数(主席树入门)

题目传送门

因为可持久化线段树和普通线段树的区别，因此我们不能像普通线段树那样按照二叉树来查找，因此我们应该使用一个指针来开点。
根据可持久化的要求，插入新版本时，先复制原节点，然后改变的节点来重新开点，不改变的节点按照原来的连接方式。
那么这道题为什么我们想到要用主席树呢？
原因还是因为我们使用的是权值线段树，会发现以值域作为区间的权值线段树，长的样子一模一样，只是节点的权值不一样，因此两颗这压根的线段树是可以相减的，但是我们这道题需要记录一个历史版本和一个全新版本，如果按照正常线段树那么空间肯定爆表。因此我们需要用主席树。
当然和扫描线类似，我们都需要离散化。

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
const int N=1e5+4;
vector<int>vec;
innt n,m,idx;
int root[N*40];
int a[N*40];
struct node{//主席树 
	int l,r;
	int cnt;
}tr[N*40];
int find(int x){
	return lower_bound(vec.begin(),vec.end(),x)-vec.begin();
}
int build(int l,int r){
	//主席树的构造 
	int p=++idx;
	if(l==r)return p;
	//如果已经遍历到子树节点
	//那么我们也无需继续递归构造主席树子节点了 
	int mid=l+r>>1;
	tr[p].l=build(l,mid);
	tr[p].r=build(mid+1,r);
	return p;
} 

int insert(int p,int l,int r,int x){//主席树插入都是子节点 
	//p:历史版本节点 ,q:我们更新的节点 
	int q=++idx;
	tr[q]=tr[p];//创建新的版本
	if(l==r){
		tr[q].cnt++;
		//找到我们要找的节点了！
		return q; 
	} 
	//没找到我们继续往下去找
	int mid=l+r>>1;
	if(x<=mid)tr[q].l=insert(tr[p].l,l,mid,x);
	else tr[q].r=insert(tr[p].r,mid+1,r,x);
	//重构一个改变的节点，不改变的节点我们不管 
	tr[q].cnt=tr[tr[q].l].cnt+tr[tr[q].r].cnt;
	return q; 
}

int query(int q,int p,int l,int r,int k){
	//p:历史版本，q:新的版本
	//在区间l和r找第k小的数字~
	if(l==r)return l;
	//对于权值线段树的增减操作
	int cnt=tr[tr[q].l].cnt-tr[tr[p].l].cnt;
	int mid=l+r>>1;
	if(k<=cnt)//就在左端点
	return query(tr[q].l,tr[p].l,l,mid,k);
	else return query(tr[q].r,tr[p].r,mid+1,r,k-cnt); 
}
int main(){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		vec.push_back(a[i]);
	}
	sort(vec.begin(),vec.end());
	vec.erase(unique(vec.begin(),vec.end()),vec.end());
	root[0]=build(0,vec.size()-1);
	for(int i=1;i<=n;i++){
		root[i]=insert(root[i-1],0,vec.size()-1,find(a[i]));
	}
	for(int i=1;i<=n;i++){
		int l,r,k;
		cin>>l>>r>>k;
		cout<<query(root[r],root[l-1],0,vec.size()-1,k)<<endl;
	}
}