hdu 4251 The Famous ICPC Team Again--划分树

[cpp]
/*
求区间中间值
可以转化为求kth值所以用了划分树
直接套用了上一篇的函数
上篇有讲解
*/
#include<iostream>
#include<algorithm>
using namespace std;
const int N=100010;
int srted[N];
struct node
{
    int num[N];
    int val[N];
}t[40];
int n,m;
void build(int l,int r,int d)//建树此树并非由节点构成而是由数组的一段构成如：d层的l~r是一个节点
{
    if(l==r) return;
    int mid=(l+r)>>1;//取中间那个
    int midd=srted[mid];
    int same=mid-l+1,samed=0,zn=l-1,yn=mid,i;//same初识化为左孩子元素个数
    //下面减去比midd小的(一定会进入左孩子里边) 剩下的就是==midd并且要进入左孩子的个数
    //samed是已经插入的数目
    //zn、yn是左右孩子的开始位置-1，下面会把元素分到两个孩子的区域里边
    for(i=l;i<=r;++i)
        if(t[d].val[i]<midd) --same;
    for(i=l;i<=r;++i)//有点像快排大的放到后边小的放前边    相等的看情况
    {
        if(i==l) t[d].num[i]=0;
        else t[d].num[i]=t[d].num[i-1];
        if(t[d].val[i]<midd)
        {
            ++t[d].num[i];//这里是统计从l到i有多少元素进入了左孩子     这是划分树主要用到的数据
            t[d+1].val[++zn]=t[d].val[i];
        }else if(t[d].val[i]>midd)//进入右孩子
        {
            t[d+1].val[++yn]=t[d].val[i];
        }else
        {
            if(samed<same)//名额还没有用完放左孩子里边
            {
                ++samed;
                ++t[d].num[i];
                t[d+1].val[++zn]=t[d].val[i];
            }else//方有孩子里边
                t[d+1].val[++yn]=t[d].val[i];
        }
    }
    build(l,mid,d+1);//建左右子树
    build(mid+1,r,d+1);
}
int query(int a,int b,int k,int l,int r,int d)//在d层[l,r]的节点里查找[a,b]中的第k大值
{
    if(a==b) return t[d].val[a];
    int mid=(l+r)>>1;
    int sx=t[d].num[a-1],sy=t[d].num[b];
    if(a-1<l) sx=0;
    if(sy-sx>=k)//[a,b]进入左子树的元素>=k
        return query(l+sx,l+sy-1,k,l,mid,d+1);
    else
    {
        int s1=(a==1?0:a-l-sx);
        int s2=(b-a+1)-(sy-sx);
        int nk=k-(sy-sx);//前(sy-sx)大的元素在左子树里剩下的在右子树里边找
        return query(mid+1+s1,mid+s1+s2,nk,mid+1,r,d+1);
    }
}
int main()
{
    int cas=1;
    int i,a,b;
    while(cin>>n)
    {
        cout<<"Case "<<cas++<<":"<<endl;
        for(i=1;i<=n;++i)
        {
            cin>>srted[i];
            t[0].val[i]=srted[i];
        }
        sort(srted+1,srted+1+n);
        build(1,n,0);
        cin>>m;
        for(i=1;i<=m;++i)
        {
            cin>>a>>b;
            cout<<query(a,b,(a+b)/2-a+1,1,n,0)<<endl;
        }
    }
}
作者：qq172108805

补充：软件开发 , C++ ,