HDU 2665 Kth number（区间第K大） (离散化+主席树)

题意就是给定一个无序数列，然后给定一堆询问，求区间[ s , t ]内的第k大的数。

查了很多资料，方法很多，看了半天只看懂了主席树怎么做，其余的做法以后补充。

主席树（函数式线段树）：

首先是离散化，将所有数按升序映射到正整数。

用线段树统计这些正整数的出现次数（Sum数组），这样就可以快速找到第k大的数。

开始时，是一颗空树，所有的Sum都为0.

用T[n]表示用数列前n个数按上述方法建立的线段树的根（以后用根来指代整棵树）。

然后要求[S,T]区间的按值建立的线段树，就只需要把两颗线段树T[R]和T[S-1]相减（因为叶节点表示的是数出现的次数，满足区间减法）。

永久化数据结构思想：

永久化数据结构思想就是在修改的时候，不修改原来数据结构，而是新建节点，并且最大限度的复用原来的数据结构。

由于T[n]和T[n-1]之间结构一样并且，只有一个叶节点的值不同，所以最多有log2 (n) 个节点不同，于是相同的节点可以复用。

就是T[n]只需要多用改变的节点的空间，其余的直接使用前一棵树的节点，这样做的好处是保存了每个时间的线段树状态，也就是保存了整个操作历史。

由于没有破坏原来的数据结构，可以访问任意时刻的线段树（从根开始递归）。要做到这点，就没办法再用原来线段树的堆式存储了，而是增加了左右子树的指针。

也可以仿照SBT的写法直接用数组表示节点，用L[]和R[]表示左右子树的下标。

建树与查询：

求T[n]的时候就是在T[n-1]中递归，然后把所有经过的节点（也就是修改过的节点）都用新节点存起来。

最后在递归访问线段树的时候，同步访问R 和 S-1两颗线段树，用Sum[R]-Sum[S-1]来表示[L,R]区间的Sum值

就可以找到第K大的数了。

还有，既然一开始是一颗空树，就不需要建立一整棵颗空树了，直接用0表示空树，省去了开始的一棵树的空间，

这样的话需要的空间就是n*log2n

所以主席树的初始化变得跟SBT一样简洁：

L[0]=R[0]=Sum[0]=T[0]=TP=0;

代码：

主席树代码：（811MS  27552KB）

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <map>
#define maxn 100007
#define maxnn  1860007
using namespace std; 
//主席树 
int L[maxnn],R[maxnn],Sum[maxnn],T[maxn],TP;//左右子树，总和，树根，指针 
void Add(int &rt,int l,int r,int x){//建立新树
	++TP;L[TP]=L[rt];R[TP]=R[rt];Sum[TP]=Sum[rt]+1;rt=TP;//复制&新建
	if(l==r) return;
	int m=(l+r)>>1;
	if(x <= m) Add(L[rt],l,m,x);
	else       Add(R[rt],m+1,r,x);
}
int Search(int TL,int TR,int l,int r,int k){//区间查询第k大
	if(l==r) return l;//返回第k大的下标
	int m=(l+r)>>1;
	if(Sum[L[TR]]-Sum[L[TL]]>=k) return Search(L[TL],L[TR],l,m,k);
	else return Search(R[TL],R[TR],m+1,r,k-Sum[L[TR]]+Sum[L[TL]]);
}
//常规
int n,m;
int value[maxn];
//离散化 
struct A{
	int x,id;
	bool operator<(const A&B)const{return x<B.x;}
}ID[maxn];
map<int,int> MP;
int Rank[maxn];
int main(void)
{
	int Test;scanf("%d",&Test);
	while(Test-->0){
		//读取输入+离散化 
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;++i) {
			scanf("%d",&value[i]);
			ID[i].x=value[i];
			ID[i].id=i;
		}
		sort(ID+1,ID+n+1);MP.clear();
		for(int i=1;i<=n;++i){
			Rank[i]=ID[i].x;
			MP[ID[i].x]=i;
		}
		//初始化主席树 
		L[0]=R[0]=Sum[0]=T[0]=TP=0;
		//建主席树 
		for(int i=1;i<=n;++i) Add(T[i]=T[i-1],1,n,MP[value[i]]);
		//开始计算 
		for(int i=0;i<m;++i){
			int s,t,k;scanf("%d%d%d",&s,&t,&k);
			printf("%d\n",Rank[Search(T[s-1],T[t],1,n,k)]); 
		}
	}
return 0;
}