poj 2104

原题链接
主席树,本题在查询前可以不用建树,每个版本加log(2e9)个节点,空间复杂度不会超过1e5 * log(2e9) 不会超过1e5 * 32,可以不用离散化直接做。

#include <iostream>

using namespace std;

typedef long long LL;

const int N = 100010;

struct Node {
    int l, r, s;
}tr[N * 32];
int n, m, idx, a[N], root[N];

int ins(int v, int l, int r, int k, int x) {
    int u = ++ idx;
    tr[u] = tr[v];
    tr[u].s ++;
    if (l == r) return u;
    LL mid = l * 1ll + r >> 1;
    if (k <= mid)   tr[u].l = ins(tr[v].l, l, mid, k, x);
    else tr[u].r = ins(tr[v].r, mid + 1, r, k, x);
    return u;
}

int query(int v, int u, int l, int r, int k) {
    if (l == r) return l;
    LL mid = l * 1ll + r >> 1;
    if (k <= tr[tr[u].l].s - tr[tr[v].l].s)   return query(tr[v].l, tr[u].l, l, mid, k);
    else return query(tr[v].r, tr[u].r, mid + 1, r, k - (tr[tr[u].l].s - tr[tr[v].l].s));
}

int main() {
    cin >> n >> m;
    idx = 0;
    for (int i = 1; i <= n; i ++) {
        int x;
        cin >> x;
        root[i] = ins(root[i - 1], -1e9, 1e9, x, 1);
    }
    while (m --) {
        int l, r, x;
        cin >> l >> r >> x;
        cout << query(root[l - 1], root[r], -1e9, 1e9, x) << endl;
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值