一道裸的划分树,求区间第k大数
第一次写,没有考虑清楚,错了无数次
错误原因:没有考虑到有很多等于中位数的数,然后在末尾有小于中位数的数,
被fxz发现了,不然还不知道要查多久呢。。最后是加了一个sm计数,记录与中位数相等的数有几个可以到左边的
划分树的思想:
也是一颗线段树,选取一个中位数,把比他小的放在左边,比他大的放在右边,原理跟快排相似,
左右两边的数的相对位置要保持不变,这样查询的时候就可以在子树上定位区间了。与中位数相等的数要特殊处理下------
有两个数组,val[i][j]表示第i层第j个元素的值,onleft[i][j]表示第i层在,在这个子树的区间[L,R]内,[L,j]中多少个数放到了左边(左子树)
建好树之后查询:
对于区间[l,r],查询第K大数,如果[l,r]区间到左边的数个数大于等于K那么应该到左子树去查找,
反之到右子树查找
到子树上查找的时候要重定位 比如 l,r区间有n个树到了左边 n<k
那么 应该到右子树查k-n大的数, l新的位置是[L,l-1]区间到右边的数的个数+mid+1,r的新位置是[L,r]区间到右边的数的个数+mid (不加1是因为mid+1是第一个位置)
反之类似
#include<algorithm>
#include<cstdio>
#include<cstring>
#define maxn 100007
using namespace std;
int ord[maxn],onleft[18][maxn],val[18][maxn];
void build(int fl,int l,int r){
if(l == r) return;
int mid = (l+r)/2,fn=fl+1;
int le=l,ri=mid+1,sm=mid+1-l;
for(int i = l;i<=r;i++)//记录左子树可以放几个等于中位数的数
if(val[fl][i]<ord[mid])sm--;
for(int i=l;i<=r;i++){
if(i==l)onleft[fl][i]=0;
else onleft[fl][i]=onleft[fl][i-1];
if(val[fl][i]<ord[mid]||(val[fl][i]==ord[mid]&&sm>0)){
if(ord[mid]==val[fl][i])sm--;
onleft[fl][i]++;
val[fn][le++]=val[fl][i];
}
else val[fn][ri++]=val[fl][i];
}
build(fl+1,l,mid),build(fl+1,mid+1,r);
}
int query(int fl,int L,int R,int l,int r,int k){
if(l == r) return val[fl][l];
int onr = onleft[fl][r],onl=0;
if(l>L) onl=onleft[fl][l-1];
int n = onr-onl,mid=(L+R)/2;
if(n >= k)return query(fl+1,L,mid,L+onl,L+onr-1,k);//左子树
else return query(fl+1,mid+1,R,mid+1+(l-L-onl),mid+1+(r-L-onr),k-n);//右子树
}
int main(){
int n, t,q;
scanf("%d",&t);
memset(onleft,0,sizeof(onleft));
while(t--){
scanf("%d%d",&n,&q);
for(int i = 1;i <= n; i++){
scanf("%d",&val[0][i]);
ord[i] = val[0][i];
}
sort(ord+1,ord+n+1);
build(0,1,n);
int l,r,k;
for(int i = 0;i < q; i++){
scanf("%d%d%d",&l,&r,&k);
printf("%d\n",query(0,1,n,l,r,k));
}
}
return 0;
}