POJ 2104（主席树）

主席树板子

题目链接：>_>

题意：询问区间第k大的值

代码：

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;

const int MAXN = 1e5 + 10;

int n, m, cnt, root[MAXN], a[MAXN], x, y, k;
struct node {
	int l, r, sum;
}T[MAXN*40];
vector<int>v;

int getid(int x) {//离散
	return lower_bound(v.begin(), v.end(), x) - v.begin() + 1;
}

void update(int l,int r,int &x,int y,int pos) {//建树，&x连接不需要更改的结点
	T[++cnt] = T[y];
	T[cnt].sum++;
	x = cnt;
	if (l == r)return;
	int mid = (l + r) / 2;
	if (mid >= pos)update(l, mid, T[x].l, T[y].l, pos);
	else update(mid + 1, r, T[x].r, T[y].r, pos);
}

int query(int l, int r, int x, int y, int k) {//查询
	if (l == r)return l;
	int mid = (l + r) / 2;
	int sum = T[T[y].l].sum - T[T[x].l].sum;
	if (sum >= k)return query(l, mid, T[x].l, T[y].l, k);
	else return query(mid + 1, r, T[x].r, T[y].r, k - sum);
}

int main() {
	scanf("%d %d", &n, &m);
	for (int i = 1; i <= n; i++) {
		scanf("%d", &a[i]);
		v.push_back(a[i]);
	}
	sort(v.begin(), v.end());
	v.erase(unique(v.begin(), v.end()), v.end());//去重
	for (int i = 1; i <= n; i++) {
		update(1, n, root[i], root[i - 1], getid(a[i]));
	}
	for (int i = 1; i <= m; i++) {
		scanf("%d %d %d", &x, &y, &k);
		printf("%d\n", v[query(1, n, root[x - 1], root[y], k) - 1]);
	}
	return 0;
}