【POI2014】KUR-Couriers 主席树

题目链接

传送门

分析

给一个长度为 n 的正整数序列 a。共有 m 组询问，每次询问一个区间 [l,r]是否存在一个数在 [l,r] 中出现的次数严格大于一半。如果存在，输出这个数，否则输出 0。

刚刚学主席树，与区间地k大类似，只需要把k变成严格大雨区间长度一半的数，分别统计小于和大于mid的数字的个数，如果这个数比k小的话就不需要继续向下访问了，由于是用vector存离散化后的结果的，要注意一下下标。

代码

#include <iostream>
#include <cstdio>
#include <cctype>
#include <vector>
#include <algorithm>

const int N = 5e5 + 10;

std::vector<int> v;
int root[N], a[N], n, m, cnt;
struct tree{
    int l, r, sum;
}t[N << 5];

inline int read(){
    int x = 0; char ch = getchar();
    while (!isdigit(ch)) ch = getchar();
    while (isdigit(ch)){x = (x<<1) + (x<<3) + (ch^48); ch = getchar();}
    return x;
}

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

inline void insert(int l, int r, int pre, int &now, int pos){
    t[++cnt] = t[pre];
    now = cnt;
    t[now].sum++;
    if (l == r) return;
    int mid = (l + r) >> 1;
    if (pos <= mid) insert(l, mid, t[pre].l, t[now].l, pos);
    else insert(mid + 1, r, t[pre].r, t[now].r, pos);
}

inline int query(int l, int r, int L, int R, int k){
    if (l == r) return l;
    int temp1 = t[t[R].l].sum - t[t[L].l].sum;
    int temp2 = t[t[R].r].sum - t[t[L].r].sum;
    int mid = (l + r) >> 1;
    if (temp1 >= k) return query(l, mid,t[L].l, t[R].l, k);
    if (temp2 >= k) return query(mid + 1, r, t[L].r, t[R].r, k);
    return 0;
}

int main() {
    n = read(), m = read();
    for (register int i = 1; i <= n; ++i)
        v.emplace_back(a[i] = read());
    std::sort(v.begin(), v.end());
    v.erase(std::unique(v.begin(), v.end()), v.end());
    for (int i = 1; i <= n; ++i) {
        insert(1, n, root[i - 1], root[i], get_id(a[i]));
    }
    while (m--){
        int l = read(), r = read();
        int len = 0;
        if ((r - l ) % 2 == 0) len = (r - l + 2) / 2;
        else len = (r - l + 3) / 2;
        int temp = query(1, n, root[l - 1], root[r], len);
        printf("%d\n", temp == 0 ? 0: v[temp - 1]);
    }
    return 0;
}