第13届景驰-埃森哲杯广东工业大学ACM程序设计大赛 C-平分游戏

第13届景驰-埃森哲杯广东工业大学ACM程序设计大赛 C-平分游戏

链接：https://www.nowcoder.com/acm/contest/90/C

来源：牛客网

题目描述

转眼间又过了一年，又有一届的师兄师姐要毕业了。

有些师兄师姐就去了景驰科技实习。

在景驰，员工是他们最宝贵的财富。只有把每一个人的专业性和独特性结合在一起，他们才会获得成功。他们致力于为所有员工打造一个能够被激励，并分享公司成功的工作环境。

创新精神：为了改变人类出行而不断迎接全新挑战。

团队协作：依靠集体的智慧，坦诚无私地帮助彼此。

结果导向：在所有方面都力争做到中国第一和世界一流，并对结果负责。

共同成长：学习永无止境，通过个人发展和职业成长实现成就。

GUDTACM集训队教练孙壕又来请大家大搓一顿。

茶余饭足以后，有人提议不如来玩游戏吧。狼人杀谁是卧底跳一跳都已经玩得太多了，所以大家决定玩一个更加有挑战性的游戏。
集训队一共有n位同学，他们都按照编号顺序坐在一个圆桌旁。第i位同学一开始有a[i]个硬币，他们希望使得每位同学手上的硬币变成相同的数目。每一秒钟，有且仅有一位同学可以把自己手上的一枚硬币交给另一位同学，其中这两位同学中间必须间隔k位同学。

现在问的是最少几秒后所有同学手上的有相同数量的硬币

输入描述:

第一行输入两个整数n,k(1<=n<=1000000,0<=k<=n)

接下来的一行有n个整数，第i个整数a[i]（0<=a[i]<=1e9）表示第i位同学手上的硬币的数量。

输出描述:

一个整数，表示最少几秒后所有同学手上的有相同数量的硬币。如果不可能，则输出gg。

示例1

输入

5 0
2 3 1 5 4

输出

3

---

思路

把会互相影响的同学放在同一个环中，就会形成很多个环，环与环之间是没有影响的，我们单独处理每个环，加起来就可以了。

例如有一个环，上面的人的硬币数是 c[1], c[2] … c[num]，这 num 个人按顺序形成了一个环，假设我们第一个人从最后一个人那里收到了x个硬币，那么当前环总时间就是：

abs(x+c[1]) + abs(x+c[1]+c[2]) + abs(x+c[1]+c[2]+c[3])…

设：b[1]=c[1], b[2]=c[1]+c[2], b[3]=c[1]+c[2]+c[3]…

显然这个可以算一下 b 数组的中位数就能得出最佳的 x 是多少，最后把所有环的答案加起来就是最终答案

注意要判断一下每个环的平均数是不是整数且相等。若 k 等于 n，意味着不能交换。

AC代码

#include <iostream>
#include <algorithm>
typedef long long ll;
const int N = 1e6+9;
using namespace std;

int a[N];
ll c[N];

void gg() {
    cout << "gg" << endl;
    exit(0);
}

int main() {
    int n,k;
    cin >> n >> k;
    int i,j;
    ll sum=0;
    for (i=1; i<=n; i++) {
        cin >> a[i];
        sum+=a[i];
    }
    if (sum%n) {
        gg();
    }
    sum/=n; // 平均值
    for (i=1; i<=n; i++) {
        a[i]-=sum;
    }
    if (k==n) { // k等于n 意味着不能交换
        for (i=1; i<=n; i++) {
            if (a[i]) {
                gg();
            }
        }
        cout << 0 << endl;
        exit(0);
    }
    int tmp;
    for (tmp=k+1; (k+1)%tmp || n%tmp; tmp--);
    //cout << "tmp=" << tmp << endl;
    ll ans=0;
    for (i=1; i<=tmp; i++) {
        int cnt=1;
        c[cnt]=a[i];
        for (j=(i+k)%n+1; j!=i; j=(j+k)%n+1) {
            cnt++;
            c[cnt]=c[cnt-1]+a[j];
        }
        if (c[cnt]) {
            gg();
        }
        sort(c+1,c+cnt+1);
        for (j=1; j<=cnt/2; j++) {
            ans-=c[j];
        }
        for (j=cnt; j>cnt-cnt/2; j--) {
            ans+=c[j];
        }
    }
    cout << ans << endl;
    return 0;
}