hdu 2665 Kth number 划分树

一道裸的划分树,求区间第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;
}


你可能感兴趣的:(数据结构,HDU,划分树)