划分树板子

划分树

划分树通过模拟快速排序的过程，每次将数据分为2部分，小的部分和大的部分，并且相同的数在序列中的位置在划分之后相对位置依然不变，通过维护num数组计算当前位置和之前的数有多少个数被分入左子树。具体教程详见博客：划分树

划分树模板

求区间第k小，同样求区间第k大可以转化为求区间[l, r]的第(r-l+1-k+1)小。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e5+100;
typedef long long ll;
#define mid (L+R)/2

int a[maxn];
int sorted[maxn];

int num[20][maxn];
int val[20][maxn];

void build(int level, int L, int R){
    if(L == R) return;
    int same = mid-L+1;
    for(int i = L; i <= R; i++) if(val[level][i] < sorted[mid]) same--;
    int ln = L, rn = mid+1;
    for(int i = L; i <= R; i++){
        if(i == L) num[level][i] = 0;
        else num[level][i] = num[level][i-1];
        if(val[level][i] < sorted[mid] || (val[level][i] == sorted[mid]&&same > 0)){
            val[level+1][ln++] = val[level][i], num[level][i]++;
            if(val[level][i] == sorted[mid]) same--;
        }
        else val[level+1][rn++] = val[level][i];
    }
    build(level+1, L, mid);
    build(level+1, mid+1, R);
}

int query(int level, int L, int R, int l, int r, int k){
    if(L == R) return val[level][L];
    int pre;
    if(l == L) pre = 0;
    else pre = num[level][l-1];
    int token = num[level][r] - pre;
    if(token >= k) return query(level+1, L, mid, L+pre, L+num[level][r]-1, k);
    else{
        int lr = mid+1 + (l-L-pre);
        return query(level+1, mid+1, R, lr, lr + (r-l+1)-token-1, k-token);
    }
}

int main(){
    int n, m;
    while(scanf("%d %d", &n, &m) != EOF){
        for(int i = 1; i <= n; i++){
            scanf("%d", &val[0][i]);
            sorted[i] = val[0][i];
        }
        sort(sorted+1, sorted+n+1);
        build(0, 1, n);
        while(m--){
            int l, r, k; scanf("%d %d %d", &l, &r, &k);
            printf("%d\n", query(0, 1, n, l, r, k));
        }
    }
    return 0;
}