P3567 [POI2014]KUR-Couriers

P3567 [POI2014]KUR-Couriers

題目:

題解:

  • 主席樹節點開個sum記錄l-r這幾個數字出現的次數。
  • 查詢的時候如果一段數的出現次數之和>區間長度/2,就往這一段數遞歸下去。當葉子節點滿足>區間長度/2時,說明有數字出現次數超過了一半。
#include <iostream>
#include <cstdio>
#include <algorithm>
#define N 500005
#define find(x) (lower_bound(b + 1, b + 1 + cnt, x) - b)
using namespace std;

struct T {int l, r, sum;} t[N << 5];
int n, m, cnt, dex;
int a[N], b[N], rt[N];

int read()
{
    int x = 0, f = 1; char c = getchar();
    while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' && c <= '9') {x = x * 10 + c - '0'; c = getchar();}
    return x *= f;
}

int build(int l, int r)
{
    int p = ++dex, mid = l + r >> 1;
    if(l == r) return p;
    t[p].l = build(l, mid), t[p].r = build(mid + 1, r);
    return p;
}

int upd(int las, int l, int r, int val)
{
    int p = ++dex, mid = l + r >> 1;
    t[p].l = t[las].l, t[p].r = t[las].r, t[p].sum = t[las].sum + 1;
    if(l == r) return p;
    if(val <= mid) t[p].l = upd(t[las].l, l, mid, val);
    else t[p].r = upd(t[las].r, mid + 1, r, val);
    return p;
}

int ask(int u, int v, int l, int r, int val)
{
    int mid = l + r >> 1, res = 0;
    if(l == r) return l;
    if(t[t[v].l].sum - t[t[u].l].sum >= val) return ask(t[u].l, t[v].l, l, mid, val);
    if(t[t[v].r].sum - t[t[u].r].sum >= val) return ask(t[u].r, t[v].r, mid + 1, r, val);
    return 0;
}

int main()
{
    cin >> n >> m;
    for(int i = 1; i <= n; i++)
        a[i] = read(), b[++cnt] = a[i];
    sort(b + 1, b + 1 + n);
    cnt = unique(b + 1, b + 1 + cnt) - b - 1;
    rt[0] = build(1, cnt);
    for(int i = 1; i <= n; i++)
        rt[i] = upd(rt[i - 1], 1, cnt, find(a[i]));
    for(int i = 1; i <= m; i++)
    {
        int l = read(), r = read();
        int id = ask(rt[l - 1], rt[r], 1, cnt, (r - l + 1) / 2 + 1);
        printf("%d\n", b[id]);
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章