K-th Number--主席树

最新推荐文章于 2021-09-07 08:14:48 发布

黄油^小饼干

最新推荐文章于 2021-09-07 08:14:48 发布

阅读量199

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/Napom/article/details/100566049

版权

算法专栏收录该内容

145 篇文章 1 订阅

订阅专栏

题目链接：https://vjudge.net/problem/POJ-2104

题目大意

给定一个序列，求区间[l, r]内第k小的数。

分析

通过这道题，对主席树有了一点点了解。

主席树的节点意义与线段树不同，主席树的每个节点存储某个前缀[1..i]中属于区间[L..R]的数一共有多少个，（比如根节点是[1..n]，一共i个数，那么sum[root] = i；根节点的左儿子是[1..(L+R)/2]，若不大于(L+R)/2的数有x个，那么sum[root.left] = x）。而且在主席树中，节点的左右孩子编号不一定满足2*i，2*i+1。一开始建一棵空树，然后把给定序列离散化，把离散化后的每个数对应的下标放入主席树中。我们可以发现，每放一个数时，只有一条链发生了改变，那么我们就只对这一条链进行操作（因为不可能每次都新建一棵树，太耗费内存）。

假设给定序列是3,1,4,2，首先建一棵空树

先放入3

再放入1

以此类推，全部放完后

注意图中连在左边的不一定是左区间，例如蓝色线段，3~4连在左边，但却是右区间。

建完树后，这些数字一定是按从小到大的顺序排列的（仔细想想就知道了）。当我们求1--r内第k小的数字时，我们先看左区间，左区间内的数大于等于k时，第k小的数一定在左区间内，小于k就在右区间内。例如，我们要求1--3内第2小的数，那我们就从绿色节点开始，先看左区间，发现1--2中只有一个数，那么第2小的数就在右区间内，注意这个时候左区间已经存在1个数了，那么在右区间内找的时候，就是要找第（2-1）小的数，也就是第1小的数，直到叶子节点为止。

这只是求1--r区间的，如果是l--r区间内第k小的数呢？主席树里面存的是前缀和，求l--r区间内时，把1--l-1区间内的减去不就行了吗。为什么可以相减？我们举个例子，假设序列是3,1,4,2，[1, 3]内第2小的数是3，[2, 3]内呢？因为3这个数不在[2, 3]这个区间内，所以我们把3这个数字产生的影响减去，答案就是4了。同时要注意建空树时，根节点编号为0，不能为1，如果为1，当l=1时，root[l-1]就不存在了。

学习博客：https://blog.csdn.net/creatorx/article/details/75446472

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1e5 + 5;
int n, q, cnt, a[N], id[N], root[N]; // root表示根节点 
struct node {
	int l, r, sum; // sum表示有几个数在该节点代表的区间内 
}tr[N * 20];
int build(int l, int r) // 建树 
{
	int cur = cnt++;
	tr[cur].sum = 0;
	if(l == r)
	{
		tr[cur].l = tr[cur].r = 0; // 叶子节点没有左右孩子 
		return cur;
	}
	int mid = (l + r) >> 1;
	tr[cur].l = build(l, mid); // 体现文中所说的左右节点编号不满足2*i，2*i+1 
	tr[cur].r = build(mid + 1, r);
	return cur;
}
int update(int up, int tar, int l, int r) // 更新 
{
	int cur = cnt++;
	tr[cur] = tr[up]; // 先让新增节点等于该节点，之后改变左或右孩子 
	tr[cur].sum++;
	if(l == r) return cur;
	int mid = (l + r) >> 1;
	if(tar <= mid) tr[cur].l = update(tr[up].l, tar, l, mid);
	else tr[cur].r = update(tr[up].r, tar, mid + 1, r);
	return cur;
}
int ask(int pl, int pr, int l, int r, int k)
{
	if(l == r) return l;
	int mid = (l + r) >> 1;
	if(tr[tr[pr].l].sum - tr[tr[pl].l].sum >= k) // 减去不在该区间内的数 
		return ask(tr[pl].l, tr[pr].l, l, mid, k);
	else
		return ask(tr[pl].r, tr[pr].r, mid + 1, r, k - (tr[tr[pr].l].sum - tr[tr[pl].l].sum)); // k发生变化 
}
int main()
{
	while(~scanf("%d %d", &n, &q))
	{
		for(int i = 1; i <= n; i++)
		{
			scanf("%d", &a[i]);
			id[i] = a[i];
		}
		// 离散化 
		sort(id + 1, id + n + 1);
		int len = unique(id + 1, id + n + 1) - (id + 1);
		cnt = 0;
		root[0] = build(1, len);
		for(int i = 1; i <= n; i++)
		{
			int p = lower_bound(id + 1, id + len + 1, a[i]) - id;
			root[i] = update(root[i - 1], p, 1, len); // 新增根节点等于新增的第一个节点 
		}
		int l, r, k;
		while(q--)
		{
			scanf("%d %d %d", &l, &r, &k);
			printf("%d\n", id[ask(root[l - 1], root[r], 1, len, k)]);
		}
	}
	return 0;
}