BestCoder Round #34_1002 以及 hdu 5191

枚举最终的W堆积木在哪，确定了区间，那么就需要把高于H的拿走，低于H的补上，高处的积木放到矮的上面，这样最优。

注意多出来的积木可以放在已有积木的前面或者后面，独立成一堆积木，所以需要在n堆积木的前后分别开一个长度为w的数组，所以整个数组的长度应该为n+2*w.

代码如下：

#include<stdio.h>
#include<string.h>

long long int a[150010],high[150010],low[150010];     //high数组用来保存前i个元素中比h高的部分的总和，low组用来保存前i个元素中比h低的部分的总和

int main()
{
    int n,w,h;

    while(scanf("%d%d%d",&n,&w,&h)!=EOF)
    {
        long long int sum=0;
        memset(a,0,sizeof(a));
        memset(high,0,sizeof(high));
        low[0]=0;                          //注意初始化
        for(int i=1;i<=w;i++)
            low[i]=i*h;
        for(int i=w+1;i<=w+n;i++)
        {
            scanf("%I64d",&a[i]);
            sum+=a[i];
            if(a[i]>h)                     //分类
            {
                high[i]=high[i-1]+a[i]-h;
                low[i]=low[i-1];
            }
            else
            {
                low[i]=low[i-1]+h-a[i];
                high[i]=high[i-1];
            }
        }
        for(int i=w+n+1;i<=n+w+w;i++)
        {
            high[i]=high[i-1];
            low[i]=low[i-1]+h;
        }
        if(sum<(long long int)w*h)
        {
            printf("-1
");
            continue;
        }
        long long int step=999999999999;
        for(int i=1;i<=n+w+1;i++)
        {
            if(high[i+w-1]-high[i-1]==low[i+w-1]-low[i-1])
            {
                if(step>high[i+w-1]-high[i-1])
                    step=high[i+w-1]-high[i-1];
            }
            if(high[i+w-1]-high[i-1]>low[i+w-1]-low[i-1])
            {
                if(step>high[i+w-1]-high[i-1])
                    step=high[i+w-1]-high[i-1];
            }
            if(high[i+w-1]-high[i-1]<low[i+w-1]-low[i-1])
            {
                if(step>low[i+w-1]-low[i-1])
                    step=low[i+w-1]-low[i-1];
            }
        }
        printf("%I64d
",step);
    }
    return 0;
}