K-th Number（主席树）

Description

链接

You are working for Macrohard company in data structures department. After failing your previous task about key insertion you were asked to write a new data structure that would be able to return quickly k-th order statistics in the array segment.
That is, given an array $a[1...n]$ of different integer numbers, your program must answer a series of questions $Q(i,j,k)$ in the form: “What would be the k-th number in $a[i...j]$ segment, if this segment was sorted?”
For example, consider the array $a=(1,5,2,6,3,7,4)$. Let the question be $Q(2,5,3)$. The segment $a[2...5]$ is $(5,2,6,3)$. If we sort this segment, we get $(2,3,5,6)$, the third number is $5$, and therefore the answer to the question is $5$.

Input

The first line of the input file contains $n$ — the size of the array, and $m$ — the number of questions to answer ($1<=n<=100000,1<=m<=5000$).
The second line contains $n$ different integer numbers not exceeding $10^9$ by their absolute values — the array for which the answers should be given.
The following $m$ lines contain question descriptions, each description consists of three numbers: $i,j$, and $k$ ($1<=i<=j<=n,1<=k<=j-i+1$) and represents the question $Q(i,j,k)$.

Output

For each question output the answer to it — the k-th number in sorted $a[i...j]$ segment.
Sample Input

7 3
1 5 2 6 3 7 4
2 5 3
4 4 1
1 7 3

Sample Output

5
6
3

Hint

This problem has huge input,so please use c-style input(scanf,printf),or you may got time limit exceed.

思路

翻译一下题意：
给出一个长度为 $n$ 的序列，有 $m$ 次询问，每次询问查询区间 $[l,r]$ 内的第 $k$ 小数。$1\le n\le 10^5，a[i]\le 10^9，m\le 5000$

首先对序列离散化。

我们对序列 $a_1$ ~ $a_i$ 建立一棵权值线段树，也即是说，离散化后的每个数字 $j$，若是出现过一次，就在对应的桶中 $cnt++$，线段树维护着这些桶。

那么，要找到 $a_1$ ~ $a_i$ 中的第 $k$ 小数，只需要使用二分的方法，查询 $a_1$ ~ $a_{\frac{1+i}{2}}$ 中的数字共出现多少次，再递归对 $a_1$ ~ $a_{\frac{1+i}{2}}$ 或 $a_{\frac{1+i}{2}+1}$ ~ $a_i$ 进行搜索，即可找到答案。

那又如何查询区间 $a_i$ ~ $a_j$ 中的第 k 小数呢？用一个类似前缀和的方法，区间 $a_i$ ~ $a_{\frac{i+j}{2}}$ 中数字出现的个数，可以通过区间 $a_1$ ~ $a_{\frac{i+j}{2}}$ 减去区间 $a_1$ ~ $a_{i-1}$来计算。这样，就可以计算任意区间的第 $k$ 小数了。

想法很不错，但是这需要对序列的任意前缀建立一棵对应的权值线段树，这样的时空复杂度可不是一般家庭能够承受的。

可持久化线段树就是这样的数据结构，一棵主席树里包含着很多棵线段树。

假设我们已经建立好了一棵线段树，维护着 $a_1$ ~ $a_i$ 这个区间。那么，如果要建另一棵线段树，维护 $a_1$ ~ $a_{i+1}$ 这个区间，其实两棵线段树之间不同的结点只有修改过的那一条链。如果能把相同的结点共用，那么就会大大降低复杂度。

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1e5+10,M=25e5;
int ls[M],rs[M],sum[M],r[M],tot;
int a[N],b[N],c[N],n,m,v,cnt;

void build(int &p,int l,int r){
	p=++tot;
	if(l==r) return;
	int mid=(l+r)>>1;
	build(ls[p],l,mid);
	build(rs[p],mid+1,r);
}

int change(int p,int l,int r){
	int q=++tot;
	ls[q]=ls[p],rs[q]=rs[p];
	sum[q]=sum[p]+1;
	if(l==r) return q;
	int mid=(l+r)>>1;
	if(v<=mid) ls[q]=change(ls[q],l,mid);
	if(v>=mid+1) rs[q]=change(rs[q],mid+1,r);
	return q;
}

int query(int i,int j,int k,int l,int r){
    if(l==r) return l;
	int mid=(l+r)>>1,x=sum[ls[j]]-sum[ls[i]];
	if(x<k) return query(rs[i],rs[j],k-x,mid+1,r);
	else return query(ls[i],ls[j],k,l,mid);
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]),b[i]=a[i];
	sort(b+1,b+n+1);
	cnt=unique(b+1,b+n+1)-b-1;
	for(int i=1;i<=n;i++)
		c[i]=lower_bound(b+1,b+cnt+1,a[i])-b;
	build(r[0],1,cnt);
	for(int i=1;i<=n;i++)
		v=c[i],r[i]=change(r[i-1],1,cnt);
	while(m--){
		int L,R,k;
		scanf("%d%d%d",&L,&R,&k);
		printf("%d\n",b[query(r[L-1],r[R],k,1,cnt)]);
	}
}