划分树板子

划分树

划分树通过模拟快速排序的过程,每次将数据分为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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值