HDU-5933-ArcSoft's Office Rearrangement

ACM模版

描述

题解

给定 N 个数要求划分为 K 份，一共有两种操作，一种是将相邻两数合并，一种是将一个数拆开两部分。

很明显的贪心模拟，这场比赛好像比较钟爱贪心模拟，可是这个题好坑，因为题目中约定的数据不可能超过 int，却挂了，必须使用 longlong 才行。

代码

#include <iostream>#include <cstdio>#include <queue>using namespace std;const int MAXN = 1e5 + 10;long long N, K;long long a[MAXN];queue<long long> qi;int main(){    int T;    scanf("%d", &T);    for (int case_ = 1; case_ <= T; case_++)    {        printf("Case #%d: ", case_);        scanf("%lld%lld", &N, &K);        long long sum = 0;        for (int i = 0; i < N; i++)        {            scanf("%lld", a + i);            sum += a[i];        }        if (sum % K)        {            puts("-1");            continue;        }        while (!qi.empty())        {            qi.pop();        }        long long ans = 0, unit = sum / K, t;        for (int i = 0; i < N; i++)        {            if (!qi.empty())            {                t = qi.front();                qi.pop();                a[i] += t;                ans++;            }            if (a[i] > unit)            {                t = a[i] / unit;                ans += t;                t = a[i] % unit;                if (t)                {                    qi.push(t);                }                else                {                    ans--;                }            }            else if (a[i] < unit)            {                qi.push(a[i]);            }        }        printf("%lld\n", ans);    }    return 0;}