第K小数——初识主席树

大体的思路和前一篇的可持久化字典树差不多，甚至可能还要简单一点，一些需要注意的地方已经在代码中标注出来了，大家可以看一下代码。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
const int N=1e5+10,M=1e4+10;
int n,m;
int a[N];
vector<int>nums;
struct node{
	int l,r;
	int cnt;//cnt表示当前子树下包含的点的个数 
}tr[N*4+N*17];//N*4是正常的线段树要开的空间，因为要保存历史版本，每一个版本最多是插入log(n)个点 
int root[N],idx;
int find(int x)
{
	return lower_bound(nums.begin(),nums.end(),x)-nums.begin();
}
//建树操作，注意这里的l和r不再表示范围，而是其左右儿子的标号 
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是上一个版本的编号，l~r是整个线段树所覆盖的区间，x是要插入的位置 
{
	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;//类似于pushup操作 
	return q; 
}
int query(int q,int p,int l,int r,int k)//如果要查询的区间是x1~x2，那q就是第x2个版本的根节点的编号，p是第x1-1个版本的根节点的编号 
{
	if(l==r)
	return r;
	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()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		nums.push_back(a[i]);
	}
	//对输入的数组进行排序和去重 
	sort(nums.begin(),nums.end()); 
	nums.erase(unique(nums.begin(),nums.end()),nums.end());
	root[0]=build(0,nums.size()-1);//这里容易忘记，在对整个主席树进行更新之前首先要建立第0个版本，至于为啥，见之前的可持久化字典树的博客 
	for(int i=1;i<=n;i++)
	root[i]=insert(root[i-1],0,nums.size()-1,find(a[i]));
	while(m--)
	{
		int l,r,k;
		scanf("%d%d%d",&l,&r,&k);
		printf("%d\n",nums[query(root[r],root[l-1],0,nums.size()-1,k)]);
	}
	return 0;
}

 YXC   NB!