Destiny--主席树

题目链接:https://vjudge.net/problem/CodeForces-840D

题目大意

n个数,q次询问,每次询问l,r区间内,出现次数严格大于(l + r)/ k的最小的数。

分析

主席树维护出现在某个区间的数的个数,查询时,要用一个ans来保存最小的下标,不能直接根据个数判断次数,要查询到叶子节点才可以。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define inf 0x3f3f3f3f
const int N = 3e5 + 5;
int n, q, cnt, ans, a[N], root[N];
struct node {
	int l, r, sum;
}tr[N * 40];
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);
	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;
}
void ask(int pl, int pr, int l, int r, int k) {
	if(l == r) {
		ans = min(ans, l);
		return ;
	} 
	int mid = (l + r) >> 1;
	if(tr[tr[pr].l].sum - tr[tr[pl].l].sum > k)
		ask(tr[pl].l, tr[pr].l, l, mid, k);
	if(tr[tr[pr].r].sum - tr[tr[pl].r].sum > k)
		ask(tr[pl].r, tr[pr].r, mid + 1, r, k);
}
int main() {
	while(~scanf("%d %d", &n, &q)) {
		cnt = 0;
		root[0] = build(1, n);
		for(int i = 1; i <= n; i++) {
			scanf("%d", &a[i]);
			root[i] = update(root[i - 1], a[i], 1, n);
		}
		int l, r, k;
		while(q--) {
			scanf("%d %d %d", &l, &r, &k);
			ans = inf;
			ask(root[l - 1], root[r], 1, n, (r - l + 1) / k);
			printf("%d\n", ans == inf ? -1 : ans);
		}
	}
	return 0;
}
发布了150 篇原创文章 · 获赞 4 · 访问量 6963

猜你喜欢

转载自blog.csdn.net/Napom/article/details/100803450