HDU 5191 Building Blocks

题目链接：

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

bc(中文):http://bestcoder.hdu.edu.cn/contests/contest_chineseproblem.php?cid=572&pid=1002

题解：

要在原始的n堆前面扩展w个空堆，同时在原始n堆后面扩展w个空堆，然后对[0,w+n+w)这个区间考虑所有的长度为w的子区间，计算出需要移动的方块数，更新答案。

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
typedef long long LL;

const int maxn = 50000+10;

int n;
LL w, h;
int arr[maxn*3];

void init() {
    memset(arr, 0, sizeof(arr));
}

int main() {
    while (scanf("%d%lld%lld", &n, &w,&h) == 3 && n) {
        init();
        LL sum = 0;
        for (int i = 0; i < n; i++) {
            scanf("%d", arr + i+w);
            sum += arr[i+w];
        }
        if (sum < w*h) {
            printf("-1\n");
            continue;
        }
        //cntf:总共需要填入多少个方块，cntz:总共需要移出多少个方块，cnt保存连续w个的初始和
        LL cntz = 0,cntf=w*h, cnt = 0;
        LL ans = w*h;
        for (int i = w; i < 2*w+n; i++) {
            //计算新窗口的cntz,cntf,cnt；
            if (arr[i - w] - h>0) cntz -= (arr[i - w] - h);
            else cntf -= (h - arr[i - w]);
            cnt -= arr[i - w];
            if (arr[i] - h > 0) cntz += arr[i] - h;
            else cntf += h - arr[i];
            cnt += arr[i];
            
            LL tmp;
            if (w*h > cnt) tmp =cntf;
            else tmp = cntz;

            ans = min(ans, tmp);
        }

        printf("%lld\n", ans);
    }
    return 0;
}