hdu 3473 划分树

题目大意:有一个数列 x1..xn,要求一个数x使得 sigma(abs(xi-x))值最小,很明显,对数列进行排序后最中间的那个数就是x,可用划分树求得,那么如何求和呢,经过计算可知,既然

x 是最中间的那个数,那么最后的和 即为 x左边 xmid-x1+xmid-x2.. x(mid+1) - xmid + x(mid+2)-xmid..  整理得 xmid*(lefnum-rignum)+rigsum-lefsum

lefnum为划分过程进入左子树的个数,lefsum为进入左子树的数之和

lefsum求法:在划分过程,当该层 有数进入左子树即 加上 该数,具体见代码。。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<math.h>
#include<algorithm>
using namespace std;

#define MAXN 100010
#define lson u<<1
#define rson u<<1|1
#define LL long long

int sorted[MAXN],val[20][MAXN],Lnum[20][MAXN];
LL sum[MAXN]={0},Lsum[20][MAXN],lefsum;
int n,q,lefnum;

struct Node{
    int lef,rig,mid;
}T[MAXN<<2];

void Build(int u,int l,int r,int d){
    T[u].lef=l;
    T[u].rig=r;
    T[u].mid=(l+r)>>1;
    if(l==r)return;
    int mid=T[u].mid;
    int lsame=mid-l+1;

    for(int i=l;i<=r;i++){
        if(val[d][i]<sorted[mid])lsame--;
    }

    int p=T[u].lef,q=T[u].mid+1,same=0;
    Lsum[d][0]=0;
    for(int i=l;i<=r;i++){
        if(i==l)Lnum[d][i]=0;
        else Lnum[d][i]=Lnum[d][i-1];
        Lsum[d][i]=Lsum[d][i-1];
        if(val[d][i]<sorted[mid]){
            val[d+1][p++]=val[d][i];
            Lsum[d][i]+=val[d][i];
            Lnum[d][i]++;
        }
        else if(val[d][i]>sorted[mid]){
            val[d+1][q++]=val[d][i];
        }
        else {
            if(same<lsame){
                val[d+1][p++]=val[d][i];
                Lsum[d][i]+=val[d][i];
                Lnum[d][i]++;
                same++;
            }
            else val[d+1][q++]=val[d][i];
        }
    }
    Build(lson,l,mid,d+1);
    Build(rson,mid+1,r,d+1);
}


int Query(int u,int l,int r,int d,int k){
    if(l==r)return val[d][l];
    else {
        int n1,n2;
        if(l==T[u].lef){n1=0;n2=Lnum[d][r];}
        else {n1=Lnum[d][l-1];n2=Lnum[d][r]-n1;}
        if(n2>=k){
            int newl=T[u].lef+n1;
            int newr=T[u].lef+n1+n2-1;
            return Query(lson,newl,newr,d+1,k);
        }
        else {
        	lefnum+=n2;
        	lefsum+=(LL)(Lsum[d][r]-Lsum[d][l-1]);
        	//printf("lefsum:%lld\n",lefsum);
            int newl=T[u].mid+1+(l-T[u].lef-n1);
            int newr=newl+(r-l+1-n2)-1;
            return Query(rson,newl,newr,d+1,k-n2);
        }
    }
}

int main(){

    int t;
    scanf("%d",&t);
    for(int cas=1;cas<=t;cas++){
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            scanf("%d",&val[0][i]);
            sorted[i]=val[0][i];
            sum[i]=sum[i-1]+val[0][i];
        }
        sort(sorted+1,sorted+1+n);
        //memset(Lsum,0,sizeof(Lsum));
        Build(1,1,n,0);
        int l,r;
        scanf("%d",&q);
        printf("Case #%d:\n",cas);
        while(q--){
            scanf("%d%d",&l,&r);
            l++;r++;
            int k=(r-l+2)>>1;
            lefnum=lefsum=0;
            int midval=Query(1,l,r,0,k);
            //printf("midval %d\n",midval);
            printf("%I64d\n",(LL)midval*(lefnum-(r-l+1-lefnum))+sum[r]-sum[l-1]-lefsum-lefsum);
            //printf("lnum%d,rnum%d,lsum%lld,rsum%lld\n",lefnum,(r-l+1-lefnum),lefsum,sum[r]-sum[l-1]-lefsum);

        }
        putchar(10);
    }

}


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值