题目链接: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;
}