主席树学习整理

主席树学习整理

1、求静态区域第k大

1.1静态主席树

发明者的原话：“对于原序列的每一个前缀[1···i]建立出一棵线段树维护值域上每个数出现的次数，则其树是可减的”

可以加减的理由：主席树的每个节点保存的是一颗线段树，维护的区间信息，结构相同，因此具有可加减性（关键）

首先开一个数组t[n]，存储内容为a中排序并去重的值（类似于离散化），每棵线段树维护的内容是a1...ai此区间中的树在t[n]中出现的次数

1.2举个栗子

an：4 1 1 2 8 9 4 4 3

将序列排序并去重后得到t[n]：

tn：1 2 3 4 8 9

对前缀a[1...9]建树，1*2，2*1，3*1，4*3，8*1，9*1，每个数出现的次数即为线段树维护的值，树中每个节点表示t[i,j],中的数字在a[1...9]中出现的次数

1.3静态主席树的一些点

1.建树时首先需要建一棵空的线段树，即最原始的主席树，此时主席树只含有一个空的节点，之后依次对原序列按某种顺序更新，就是将原序列加入到相应的位置

2.主席树是一种特殊的线段树集，它包含了所有线段树的优势，并且可以保存历史状态，主席树查找和更新的时空复杂度都为O(nlogn)且总空间复杂度为O(nlogn+nlogn)前者为空树的复杂度，后者为更新n次的空间复杂度，缺点是空间损耗巨大

3.主席树可以处理区间[L,R]中介于[x,y]的值的问题

4.若增加空间垃圾回收则可以使空间复杂度降低一个log

1.4代码

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#pragma warning (disable:4996)
const int maxn = 200005;
int n, m, tsize;
int a[maxn], b[maxn], t[maxn], tot = 0;
struct charitree{
	int left, right, size;
}tree[maxn * 30];
int build(int stdl, int stdr) {
	int root = tot++;
	tree[root].size = 0;
	if (stdl == stdr)
		return root;
	int mid = (stdl + stdr) >> 1;
	tree[root].left = build(stdl, mid);
	tree[root].right = build(mid + 1, stdr);
	return root;
}
int update(int root, int x) {
	int now = tot++;
	int tmp = now;
	tree[now].size = tree[root].size + 1;
	int stdl = 1, stdr = tsize;
	while (stdl < stdr) {
		int mid = (stdl + stdr) >> 1;
		if (x <= mid) {
			tree[now].left = tot++;
			tree[now].right = tree[root].right;
			now = tree[now].left;
			stdr = mid;
			root = tree[root].left;
		}
		else {
			tree[now].left = tree[root].left;
			tree[now].right = tot++;
			now = tree[now].right;
			stdl = mid + 1;
			root = tree[root].right;
		}
		tree[now].size = tree[root].size + 1;
	}
	return tmp;
}
int ask(int stdl, int stdr, int k) {
	int l = 1, r = tsize;
	while (l<r)
	{
		int mid = (l + r) >> 1;
		if (tree[tree[stdr].left].size - tree[tree[stdl].left].size >= k) {
			r = mid;
			stdl = tree[stdl].left;
			stdr = tree[stdr].left;
		}
		else {
			l = mid + 1;
			k -= tree[tree[stdr].left].size - tree[tree[stdl].left].size;
			stdl = tree[stdl].right;
			stdr = tree[stdr].right;
		}
	}
	return l;
}
int find(int x) {
	return lower_bound(b + 1, b + 1 + tsize, x) - b;
}
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 + 1 + n);
	tsize = unique(b + 1, b + 1 + n) - b - 1;
	t[0] = build(1, tsize);
	for (int i = 1; i <= n; i++) {
		t[i] = update(t[i - 1], find(a[i]));
	}
	int l, r, k;
	while (m--)
	{
		scanf("%d%d%d", &l, &r, &k);
		printf("%d\n", b[ask(t[l - 1], t[r], k)]);
	}
	return 0;
}