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