codeforces840D Destiny (主席树求区间＞=num的最小值)

题目链接: Destiny

大致题意

有一个下标从1~n的序列, 每次给出一组询问 l, r, k. 询问在[l, r]区间, 是否有元素出现次数 > (r - l + 1) / k. 如果存在的话, 输出最小的元素, 反之输出-1.

解题思路

我们可以通过主席树来实现.

具体做法是, 在权值线段树上每次比较左半区间和右半区间的元素个数之和是否满足条件, 若满足则进行递归求解.

考虑到复杂度问题, 如果假设k == 2的情况, 我们很容易观测到, 左半区间和右半区间只会满足一个, 不可能同时满足. 因此复杂度为logn. 而当k更大时, 其实我们可以把上面情况推广, 即: 不会存在大于等于k个数字同时满足条件. 因为要查询大于(r - l + 1) / k的数, 若有k个同时满足, 则区间元素个数会超过区间长度, 很显然是不成立的.

因此单次查询最坏复杂度为: O(klogn)

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
#define debug(a) cout << #a << " = " << a << endl;
using namespace std;
typedef long long ll;
const int N = 3E5 + 10, INF = 0x3f3f3f3f;
struct node {
    int l, r;
    int cou;
}t[N << 5];
int root[N], ind;
int build(int a, int c, int tl, int tr, int p) {
    int x = ++ind;
    t[x] = t[p]; t[x].cou += c;
    if (tl == tr) return x;
    int mid = tl + tr >> 1;
    if (a <= mid) t[x].l = build(a, c, tl, mid, t[p].l);
    else t[x].r = build(a, c, mid + 1, tr, t[p].r);
    return x;
}

int ask(int num, int tl, int tr, int p, int x) {
    if (tl == tr) {
        if (t[x].cou - t[p].cou >= num) return tl;
        return INF; 
    }
    int mid = tl + tr >> 1;
    int coul = t[t[x].l].cou - t[t[p].l].cou, cour = t[t[x].r].cou - t[t[p].r].cou;

    int res = INF;
    if (coul >= num) res = ask(num, tl, mid, t[p].l, t[x].l);
    if (cour >= num) res = min(res, ask(num, mid + 1, tr, t[p].r, t[x].r));
    return res;
}
int main()
{
    int n, m; cin >> n >> m;
    rep(i, n) {
        int x; scanf("%d", &x);
        root[i] = build(x, 1, 1, n, root[i - 1]);
    }

    while (m--) {
        int l, r, k; scanf("%d %d %d", &l, &r, &k);
        int need = (r - l + 1) / k + 1;
        int res = ask(need, 1, n, root[l - 1], root[r]);
        printf("%d\n", res == INF ? -1 : res);
    }
    return 0;
}

END