vijos P1081野生动物园 主席树求区间第K大

 

P1081野生动物园

描述

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 2
1 5 2 6 3 7 4
1 5 3
2 7 1

样例输出1[复制]

3
2

限制

各个测试点2s

提示

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

第一次“抄”主席树。。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

const int M = 1000010;

struct Node{
    int l, r, sum;
}T[M*20];
int T_cnt;
void inser(int &num, int &x, int L, int R) {
     T[T_cnt++] = T[x]; x = T_cnt - 1;
     ++T[x].sum;
     if(L == R) return ;
     int mid = (L + R) >> 1;
     if(num <= mid) inser(num, T[x].l, L, mid);
     else inser(num, T[x].r, mid + 1, R);
 }
int query(int i, int j, int k, int L, int R) {
     if(L == R) return L;
     int t = T[T[j].l].sum - T[T[i].l].sum;
     int mid = (R + L) >> 1;
     if(k <= t) return query(T[i].l, T[j].l, k, L, mid);
     else return query(T[i].r, T[j].r, k - t, mid + 1, R);
 }
struct A{
    int x, idx;
    bool operator < (const A & o)const{
        return x < o.x;
    }
}a[M];
int ranks[M], root[M];
int n, m;
int main() {
    T[0].l = T[0].r = T[0].sum = 0;
    root[0] = 0;
    while(~scanf("%d%d", &n, &m)){
        for(int i = 1; i <= n; i++){
            scanf("%d", &a[i].x);
            a[i].idx = i;
        }
        sort(a+1, a+1+n);
        for(int i = 1; i <= n; i++) ranks[a[i].idx] = i;
        T_cnt = 1;
        for(int i = 1; i <= n; i++){
            root[i] = root[i-1];
            inser(ranks[i], root[i], 1, n);
        }
       //printf("%SSS\n");
        while(m--){
            int i, j, k;
            scanf("%d%d%d", &i, &j, &k);
            printf("%d\n", a[query(root[i-1], root[j], k, 1, n)].x);
        }
    }
    return 0;
}