欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

主席树 —— 区间第K大

程序员文章站 2024-03-02 22:52:04
...

模板题:POJ 2104

区间第K大模板:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
//using pii = pair <ll,int>;
const int maxn = 1e5 + 10;
int n, m, a[maxn], root[maxn], tot;
int t[maxn*30], ls[maxn*30], rs[maxn*30];
vector <int> v;

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

void update(int &rt, int pre, int l, int r, int pos){
	if(l>pos || r<pos) return;
	rt = ++tot;
	t[rt] = t[pre] + 1;
	ls[rt] = ls[pre], rs[rt] = rs[pre];
	if(l == r) return;
	int m = l + r >> 1;
	update(ls[rt],ls[pre],l,m,pos);
	update(rs[rt],rs[pre],m+1,r,pos);
}

int query(int rt, int pre, int l, int r, int k){
	if(l == r) return l;
	int m = l + r >> 1;
	int sum = t[ls[rt]] - t[ls[pre]];
	if(sum >= k) return query(ls[rt],ls[pre],l,m,k);
	return query(rs[rt],rs[pre],m+1,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(root[i], root[i-1], 1, n, getid(a[i]));
	for(int i=1; i<=m; i++){
		int l, r, k;
		scanf("%d%d%d", &l, &r, &k);
		printf("%d\n", v[query(root[r], root[l-1], 1, n, k) - 1]);
	}
}