vijos - P1081野生动物园 (划分树 + 线段树)

P1081野生动物园

Accepted

标签：[显示标签]

描述

cjBBteam拥有一个很大的野生动物园。这个动物园坐落在一个狭长的山谷内，这个区域从南到北被划分成N个区域，每个区域都饲养着一头狮子。这些狮子从北到南编号为1,2,3,…,N。每头狮子都有一个觅食能力值Ai，Ai越小觅食能力越强。饲养员cmdButtons决定对狮子进行M次投喂，每次投喂都选择一个区间[I,J]，从中选取觅食能力值第K强的狮子进行投喂。值得注意的是，cmdButtons不愿意对某些区域进行过多的投喂，他认为这样有悖公平。因此cmdButtons的投喂区间是互不包含的。你的任务就是算出每次投喂后，食物被哪头狮子吃掉了。

格式

输入格式

输入第一行有两个数N和M。此后一行有N个数，从南到北描述狮子的觅食能力值。此后M行，每行描述一次投喂。第t+2的三个数I,J,K表示在第t次投喂中，cmdButtons选择了区间[I,J]内觅食能力值第K强的狮子进行投喂。

输出格式

输出有M行，每行一个整数。第i行的整数表示在第i次投喂中吃到食物的狮子的觅食能力值。

样例1

样例输入1[复制]

7 21 5 2 6 3 7 41 5 32 7 1

样例输出1[复制]

32

限制

各个测试点2s

提示

对于100%的数据，有1<=N<=100000，1<=M<=50000。

来源

周戈林

#include <map>#include <set>#include <cmath>#include <ctime>#include <queue>#include <vector>#include <cctype>#include <cstdio>#include <string>#include <cstring>#include <sstream>#include <cstdlib>#include <iostream>#include <algorithm>#include <functional>using namespace std;#define pb push_back#define mp make_pair#define fillchar(a, x) memset(a, x, sizeof(a))#define copy(a, b) memcpy(a, b, sizeof(a))typedef long long LL;typedef pair<int, int > PII;typedef unsigned long long uLL;template<typename T>void print(T* p, T* q, string Gap = " ") {    int d = p < q ? 1 : -1;    while(p != q) {        cout << *p;        p += d;        if(p != q) cout << Gap;    }    cout << endl;}template<typename T>void print(const T &a, string bes = "") {    int len = bes.length();    if(len >= 2)cout << bes[0] << a << bes[1] << endl;    else cout << a << endl;}const int INF = 0x3f3f3f3f;const int MAXM = 1e5;const int MAXN = 1e5 + 5;LL sum[20][MAXN];int tree[20][MAXN], sorted[MAXN];int N, M, I, J, K;void build(int rt, int l, int r) {    int mid = (l + r) >> 1, p = l, q = mid + 1, same = mid - l + 1;//得到应该平分为二的大小    for(int i = l; i <= mid ; i ++) {//小于的话先去掉        if(sorted[i] < sorted[mid]) same --;    }    for(int i = l; i <= r; i ++) {        if(i == l) {            sum[rt][i] = 0;        } else {            sum[rt][i] = sum[rt][i - 1];        }        if(tree[rt][i] == sorted[mid]) {            if(same) {//如果相同就需要特殊判断相同的数字分在左边还是右边                same --;                sum[rt][i] ++;                tree[rt + 1][p ++] = tree[rt][i];            } else {                tree[rt + 1][q ++] = tree[rt][i];            }        } else if(tree[rt][i] < sorted[mid]) {            sum[rt][i] ++;            tree[rt + 1][p ++] = tree[rt][i];        } else {            tree[rt + 1][q ++] = tree[rt][i];        }    }    if(l == r) return;    build(rt + 1, l, mid);    build(rt + 1, mid + 1, r);}int query(int rt,int l, int r, int L, int R, int k) {    int s, ss, mid = (l + r) >> 1;    if(l == r) return tree[rt][l];    if(l == L) {        s = 0, ss = sum[rt][R];    } else {        s = sum[rt][L - 1];//得到[l, L - 1]在左边的大小        ss = sum[rt][R] - s;//得到[L, R]在左边的大小    }    if(k <= ss) {        return query(rt + 1, l, mid, l + s, l + s + ss - 1, k);//如果分在左边的话，取l + 左边的数    } else {        return query(rt + 1, mid + 1, r, mid + 1 + L - l - s, mid + 1 + R - l - s - ss, k - ss);//如果分在右边的话取mid + 1 + 分在右边的数    }}int main() {    while(cin >> N >> M) {        for(int i = 1 ; i <= N ; i ++) {            cin >> sorted[i];            tree[0][i] = sorted[i];        }        sort(sorted + 1, sorted + N + 1);        build(0, 1, N);        while(M --) {            cin >> I >> J >> K;            print(query(0, 1, N, I, J, K));        }    }    return 0;}