划分树——求区间第k小值
题目:K-th Number
描述
您正在为数据结构部门的Macrohard公司工作。在关于密钥插入的上一个任务失败后,您被要求编写一个新的数据结构,该结构能够快速返回数组段中的第k个顺序统计信息。
也就是说,给定一个不同整数的[1 ... n]数组,你的程序必须以下列形式回答一系列问题Q(i,j,k):“什么是第k个数字[i ... j]段,如果此段已分类?”
例如,考虑数组a =(1,5,2,6,3,7,4)。让问题是Q(2,5,3)。段a [2 ... 5]是(5,2,6,3)。如果我们对这个段进行排序,我们得到(2,3,5,6),第三个数字是5,因此问题的答案是5。
输入
输入文件的第一行包含n ---数组的大小,m ---要回答的问题数
(1 <= n <= 100 000,1 <= m <= 5 000)。
第二行包含n个不同的整数,它们的绝对值不超过1e9 ---应该给出答案的数组。
以下m行包含问题描述,每个描述由三个数字组成:i,j和k(1 <= i <= j <= n,1 <= k <= j - i + 1)并表示问题Q (i,j,k)。
输出
对于每个问题输出它的答案---排序的[i ... j]段中的第k个数字。
样本输入
7 3
1 5 2 6 3 7 4
2 5 3
4 4 1
1 7 3
样本输出
5
6
3
题意大概就是给一个数组,然后给出一个区间和k,求区间内第k小的值。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define MAXN 100010
#define lson l,m,d+1
#define rson m+1,r,d+1
int a[20][MAXN],sorted[MAXN],toLeft[20][MAXN];
void build(int l,int r,int d) {
if(l==r) return ;
int m=(l+r)>>1,p1=l-1,p2=m;
int mv=sorted[m];
if(l==1) toLeft[d][0]=0;
int lsame=m-l+1;
for(int i=l;i<=r;++i) if(a[d][i]<mv) --lsame;//找出左子树可容纳多少中位数
for(int i=l;i<=r;++i) {
toLeft[d][i]=toLeft[d][i-1];
if(a[d][i]<mv) ++toLeft[d][i],a[d+1][++p1]=a[d][i];
else if(a[d][i]>mv) a[d+1][++p2]=a[d][i];
else {
if(lsame) ++toLeft[d][i],a[d+1][++p1]=a[d][i],--lsame;//左子树还可容纳中位数
else a[d+1][++p2]=a[d][i];
}
}
build(lson);
build(rson);
}
int query(int k,int x,int y,int l,int r,int d) {
if(l==r) return sorted[l];
int l1=toLeft[d][x-1]-toLeft[d][l-1],l2=toLeft[d][y]-toLeft[d][x-1]; //l1=[l,x-1]上划入左子树的数目,l2=[x,y]上划入左子树的数目
int m=(l+r)>>1;
int r1=x-l-l1,r2=y-x+1-l2; //r1=[l,x-1]上划入右子树的数目,r2=[x,y]上划入右子树的数目
if(k<=l2) return query(k,l+l1,l+l1+l2-1,lson);
else return query(k-l2,m+1+r1,m+r1+r2,rson);
}
int main() {
ios::sync_with_stdio(false);
int n,m;
cin >> n >> m;
for(int i=1;i<=n;++i) cin >> a[1][i],sorted[i]=a[1][i];
sort(sorted+1,sorted+1+n);
build(1,n,1);
while(m--) {
int a,b,k;
cin >> a >> b >> k;
cout << query(k,a,b,1,n,1) << endl;
}
return 0;
}