HDU 3473 Minimum Sum 划分树,数据结构 难度:1

http://acm.hdu.edu.cn/showproblem.php?pid=3473

划分树模板题目,需要注意的是划分树的k是由1开始的

划分树:

参考:http://blog.csdn.net/shiqi_614/article/details/8041390

划分树的定义

划分树定义为，它的每一个节点保存区间[lft,rht]所有元素，元素顺序与原数组（输入）相同，但是，两个子树的元素为该节点所有元素排序后(rht-lft+1)/2个进入左子树，其余的到右子树，同时维护一个num域，num[i]表示lft->i这个点有多少个进入了左子树。

划分树的Sample

#include <cstdio>

#include <cstring>

#include <algorithm>

using namespace std;

typedef long long ll;

const int maxn=1e5+5;

const int maxf=20;

int order[maxn];

int v[maxf][maxn];

ll s[maxn];

ll sum[maxf][maxn],asum;

int num[maxf][maxn];

void build(int l,int r,int ind){

        if(l==r)return ;

        int mid=(l+r)>>1;

        int ln=l,rn=mid+1,same=mid-l+1;

        for(int i=l;i<=r;i++){

                if(v[ind][i]<order[mid])same--;

        }

        for(int i=l;i<=r;i++){

                int flag=0;

                if(v[ind][i]==order[mid]&&same>0){

                        same--;

                        flag=1;

                        v[ind+1][ln++]=v[ind][i];

                        sum[ind][i]=(i>0?sum[ind][i-1]:0)+v[ind][i];

                }

                else if(v[ind][i]<order[mid]){

                        flag=1;

                        v[ind+1][ln++]=v[ind][i];

                        sum[ind][i]=(i>0?sum[ind][i-1]:0)+v[ind][i];

                }

                else {

                        sum[ind][i]=i>0?sum[ind][i-1]:0;

                        v[ind+1][rn++]=v[ind][i];

                }

                num[ind][i]=(i>0?num[ind][i-1]:0)+flag;

        }

        build(l,mid,ind+1);

        build(mid+1,r,ind+1);

}

int query(int s,int e,int k,int l,int r,int ind){

        if(l==r)return v[ind][l];

        int mid=(l+r)>>1;

        int lls=num[ind][s-1]-num[ind][l-1];

        int lse=num[ind][e]-num[ind][s-1];

        int rls=s-l-lls;

        int rse=e-s-lse+1;

        if(lse>=k)return query(l+lls,l+lls+lse-1,k,l,mid,ind+1);

        asum+=sum[ind][e]-(s>0?sum[ind][s-1]:0);

        return  query(mid+1+rls,mid+rls+rse,k-lse,mid+1,r,ind+1);

}

int main(){

        int T;

        scanf("%d",&T);

        for(int t=1;t<=T;t++){

                int n;

                scanf("%d",&n);

                for(int i=0;i<n;i++){

                        scanf("%d",&v[0][i]);

                        order[i]=v[0][i];

                        s[i]=(i>0?s[i-1]:0)+v[0][i];

                }

                sort(order,order+n);

                memset(num,0,sizeof(num));

                build(0,n-1,0);

                int q;

                scanf("%d",&q);

                printf("Case #%d:\n",t);

                for(int i=0;i<q;i++){

                        int l,r;

                        scanf("%d%d",&l,&r);

                        asum=0;

                        ll mid=query(l,r,((l+r)>>1)-l+1,0,n-1,0);

                        ll ans=mid*(((l+r)>>1)-l)-(r-((l+r)>>1))*mid-asum+(s[r]-(l>0?s[l-1]:0)-asum-mid);

                        printf("%I64d\n",ans);

                }

                puts("");

        }

}