[HDOJ5933]ArcSoft's Office Rearrangement（贪心）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5933

题意：长度为nn的数组: a_1, a_2, \cdotsa​1​​,a​2​​,⋯, 每次操作要么可以merge两个相邻的数为一个, 值为两个数的和; 要么可以把一个数分裂成两个, 两个数的和为原数. 用最少的操作把数组变换成长度为KK且所有数值相等的数组, 无解输出-1−1。

读错题了啊…以为是任意merge，题意是相邻的merge。

 

分情况讨论，每次判断当前点是比avg大还是小，如果大的话就把多余的部分放到下一个点上，拆操作是商-1次。余数merge到下一个的操作是2次（一次拆，一次merge），如果小于平均的话就全merge到上一个去。 当然考虑当前和下一个的和能不能作为一次。

 

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
const int maxn = 100100;
int n, k;
LL a[maxn];
LL avg;

int main() {
    // freopen("in", "r", stdin);
    int T, _ = 1;
    scanf("%d", &T);
    while(T--) {
        scanf("%d%d",&n,&k);
        printf("Case #%d: ", _++);
        avg = 0;
        LL s = 0;
        for(int i = 1; i <= n; i++) {
            scanf("%d", &a[i]);
            s += a[i];
        }
        if(s % k != 0) {
            puts("-1");
            continue;
        }
        avg = s / k;
        LL ret = 0;
        for(int i = 1; i <= n; i++) {
            if(a[i] < avg) {
                if(a[i] + a[i+1] <= avg) {
                    ret++;
                    a[i+1] = a[i] + a[i+1];
                }
                else {
                    a[i+1] = a[i+1] - avg + a[i];
                    ret += 2;
                }
            }
            else if(a[i] > avg) {
                LL p = a[i] / avg;
                LL q = a[i] % avg;
                if(q != 0) {
                    a[i+1] += q;
                    ret += 2;
                }
                ret += p - 1;
            }
        }
        printf("%I64d\n", ret);
    }
    return 0;
}