读书人

hdu 3473 区划树

发布时间: 2012-11-03 10:57:43 作者: rapoo

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 longint 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);    }}


读书人网 >编程

热点推荐