【Luogu】静态区间第 k 小问题（可持续化线段树）

目录

1001. 静态区间第 k 小问题

限制条件

题目描述

输入格式

输出格式

样例

解题步骤 

1.离散化

将得到的数组，转换为它在去重后的排名。

如一个数组a = {10,24,21,22,10}变为Hash = {1,4,2,3,1};

2.建树（build）

注意这里线段树的节点表示在第k代线段树中[l,r]区间出现了几个数字。（[即当前这一颗线段树的数字在l到r之间的有几个）

3.建新版本线段树

警示后人：一定要先让新建节点的val变为原节点的val在+1

错误示范代码：

4.求出第k大的数。

5.AC代码

6.记录一下写时容易犯的错误

完

---

洛谷上题目的链接： 

1001. 静态区间第 k 小问题

限制条件

时间限制: 2000 ms, 空间限制: 256 MB

题目描述

给定N个整数构成的序列，将对于指定的闭区间查询其区间内的第K小值

输入格式

第一行包含两个正整数N、M，分别表示序列的长度和查询的个数。

第二行包含N个整数，表示这个序列各项的数字。

接下来M行每行包含三个整数l,r,k, 表示查询区间[l,r]内的第k小值。

输出格式

输出包含k行，每行1个整数，依次表示每一次查询的结果

样例

输入样例  复制

5 5
25957 6405 15770 26287 26465 
2 2 1
3 4 1
4 5 1
1 2 2
4 4 1

输出样例  复制

---

6405
15770
26287
25957
26287

解题步骤 

1.离散化

将得到的数组，转换为它在去重后的排名。

如一个数组a = {10,24,21,22,10}变为Hash = {1,4,2,3,1};

并再次基础上建一个ha数组。ha[i]表示第i名实际上的数字。

2.建树（build）

用可持续化线段树的方式来建线段树。如果你不会，请看我的这一篇博客。

先建一颗空树。

注意这里线段树的节点表示在第k代线段树中[l,r]区间出现了几个数字。（[即当前这一颗线段树的数字在l到r之间的有几个）

代码

int build(int begin,int end){
	if(begin == end){
		tot++;
		return tot;
	}
	tot++;
	int u = tot,mid = (begin+end)/2;
	sgt[u].lc = build(begin,mid);
	sgt[u].rc = build(mid+1,end);
	return u;
}//建一颗空树 

3.建新版本线段树

 对离散化数组按原序建立出每一个版本。

核心代码

int ins(int index,int begin,int end,int x){
	if(begin == end){
		tot++;
		sgt[tot].val = sgt[index].val+1;
		return tot;
	}
	tot++;
	int u = tot,mid = (begin+end)/2;
	sgt[u].val = sgt[index].val;
	sgt[u].val++;
	if(x <= mid){
		sgt[u].lc = ins(sgt[index].lc,begin,mid,x);
		sgt[u].rc = sgt[index].rc;
	}
	else{
		sgt[u].lc = sgt[index].lc;
	    sgt[u].rc = ins(sgt[index].rc,mid+1,end,x);	
	}
	return u;
} 

警示后人：一定要先让新建节点的val变为原节点的val在+1

错误示范代码：

int ins(int index,int begin,int end,int x){
	if(begin == end){
		tot++;
		sgt[tot].val++;//错误1
		return tot;
	}
	tot++;
	int u = tot,mid = (begin+end)/2;
	sgt[u].val++;//错误2
	if(x <= mid){
		sgt[u].lc = ins(sgt[index].lc,begin,mid,x);
		sgt[u].rc = sgt[index].rc;
	}
	else{
		sgt[u].lc = sgt[index].lc;
	    sgt[u].rc = ins(sgt[index].rc,mid+1,end,x);	
	}
	return u;
} 

4.求出第k大的数。

利用前缀和的思想。因为第k个版本表示了插入第k个数时的情况，所以他们（r和l-1）对应的节点相减一定是一定是在[l,r]区间中【begin~end】的数字出现的次数。

所以我们就求出他的左子树的字出现的次数。

若x >= k就往右找，反之则在排除了x的影响（即k = k-x）后往右找

为什么要排除x的影响？

因为[mid+1,end]改区间不包括左子树上的出现次数，那一边已经被排除了（类似二叉搜索树的解释）

核心代码

int gets(int u,int v,int l,int r,int k){
    if(l == r)return l;
    int x = sgt[sgt[v].lc].val-sgt[sgt[u].lc].val;
    int mid = (l+r)/2;
    if(x >= k)return gets(sgt[u].lc,sgt[v].lc,l,mid,k);
    else return gets(sgt[u].rc,sgt[v].rc,mid+1,r,k-x);
}

5.AC代码

经过以上分析我们就可以得出代码

#include<bits/stdc++.h>
using namespace std;
int a[1000100],b[1000100];
map<int,int> Hash,ha;
struct kkk{
	int val,lc,rc;
}sgt[80008000];
int ver[1000100];
int tot; 
int build(int begin,int end){
	if(begin == end){
		tot++;
		return tot;
	}
	tot++;
	int u = tot,mid = (begin+end)/2;
	sgt[u].lc = build(begin,mid);
	sgt[u].rc = build(mid+1,end);
	return u;
}//建一颗空树 
int ins(int index,int begin,int end,int x){
	if(begin == end){
		tot++;
		sgt[tot].val = sgt[index].val+1;
		return tot;
	}
	tot++;
	int u = tot,mid = (begin+end)/2;
	sgt[u].val = sgt[index].val;
	sgt[u].val++;
	if(x <= mid){
		sgt[u].lc = ins(sgt[index].lc,begin,mid,x);
		sgt[u].rc = sgt[index].rc;
	}
	else{
		sgt[u].lc = sgt[index].lc;
	    sgt[u].rc = ins(sgt[index].rc,mid+1,end,x);	
	}
	return u;
} 
int gets(int u,int v,int l,int r,int k){
    if(l == r)return l;
    int x = sgt[sgt[v].lc].val-sgt[sgt[u].lc].val;
    int mid = (l+r)/2;
    if(x >= k)return gets(sgt[u].lc,sgt[v].lc,l,mid,k);
    else return gets(sgt[u].rc,sgt[v].rc,mid+1,r,k-x);
}
int main(){
	int n,m;
	cin>>n>>m;
	for(int i = 1;i <= n;i++){
		cin>>a[i];
		b[i] = a[i];//记忆a数组 
	}
	sort(a+1,a+1+n);//排序进行排名；
	int cnt = 0;//表示当前排名 
	for(int i = 1;i <= n;i++){
		if(i == 1||a[i] != a[i-1]){//去重
		    cnt++;
			Hash[a[i]] = cnt;
			ha[cnt] = a[i]; 
		}
	} //进行离散化 
	ver[0] = 1;
	build(1,n);
	for(int i = 1;i <= n;i++){
		ver[i] = ins(ver[i-1],1,n,Hash[b[i]]);
	}
	while(m--){
		int a,b,k;
		cin>>a>>b>>k;
		cout<<ha[gets(ver[a-1],ver[b],1,n,k)]<<endl;
	}
} 

6.记录一下写时容易犯的错误

1.代码中原节点的值是不会变的。不要写成

sgt[index].val++

2.Hash一定要用map不然会RE

3.一定要先让新建节点的val变为原节点的val+1

完