HDU 2665 Kth number 划分树

题目大意:给定一个序列,求区间第k小

注意是第k小!!别被题目描述骗到了!!这题求的是第k小!!不是第k大!!!

这题和POJ2104一样,都是求区间第k小,不同的是这题的序列是有重复的

对于有重复的,我们必须先预处理出有多少个中位数能进入左区间,否则就会导致过多的中位数堆积在左区间导致该进入左区间的东西被硬塞进了右区间

其实我只是想说为何网上的处理重复都写的那么麻烦。。。像我这样精简点不好么?

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 100100
using namespace std;
int n,m,a[M],b[M],c[M];
int s[20][M];
void Build_Tree(int l,int r,int dpt)
{
    int i,mid=l+r>>1;
    if(l==r)
        return ;
    int l1=l,l2=mid+1,left=mid-l+1;
    for(i=l;i<=r;i++)
        left-=(a[i]<c[mid]);
    for(i=l;i<=r;i++)
    {
        if(a[i]>c[mid]||a[i]==c[mid]&&!left)
            b[l2++]=a[i],s[dpt][i]=(i==l?0:s[dpt][i-1]  );
        else
            b[l1++]=a[i],s[dpt][i]=(i==l?1:s[dpt][i-1]+1),left-=(a[i]==c[mid]);
    }
    memcpy( a+l , b+l , sizeof(a[0])*(r-l+1) );
    Build_Tree(l,mid,dpt+1);
    Build_Tree(mid+1,r,dpt+1);
}
int getans(int l,int r,int dpt,int x,int y,int k)
{
    int mid=l+r>>1;
    if(l==r)
        return c[mid];
    int l1=(x==l?0:s[dpt][x-1]),l2=s[dpt][y];
    if(k<=l2-l1)
        return getans(l,mid,dpt+1,l+l1,l+l2-1,k);
    else
        return getans(mid+1,r,dpt+1,(mid+1)+(x-l-l1),(mid+1)+(y-l+1-l2)-1,k-l2+l1);
}
int main()
{
    int T,i,x,y,k;
    for(cin>>T;T;T--)
    {
        cin>>n>>m;
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]),c[i]=a[i];
        sort(c+1,c+n+1);
        Build_Tree(1,n,0);
        for(i=1;i<=m;i++)
            scanf("%d%d%d",&x,&y,&k),printf("%d\n", getans(1,n,0,x,y,k ) );
    }
}


你可能感兴趣的:(HDU,划分树,hdu2665)