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

C - 平分游戏

思路：先看怎么处理一个圈有$n$个人只给相邻的人硬币的解法，保证总数整除$n$是必然的，有个很明显的情况就是如果$A$给了$B$，那么$B$将不会再给$A$，这样无疑是多余的步骤，所以相邻的两个人$AB$之间要么是$A$给$B$，要么是$B$给$A$，那么相邻的第$i-1$，$i$，$i+1$三个人，不妨假设第$i$个人给了第$i-1$个人$x_{i}$个硬币，从第$i+1$个人中拿到了$x_{i+1}$个硬币，其中$x_{i-1}$可以是负数，代表是$i-1$给$i$硬币，假设最终每个人有$M$个硬币，初始状态第$i$个人有$a_{i}$个硬币，那么有以下等式：
对于第一个人，$a_{1}-x_{1}+x_{2}=M$ $\Rightarrow$ $x_{2}= M - a_{1}+x_{1}=x_{1} - C_{1}(C_{1} = a_{1} - M,下面类似)$
对于第二个人，$a_{2}-x_{2}+x_{3}=M$ $\Rightarrow$ $x_{3}= M - a_{2}+x_{2}=2M - a_{1}- a_{2}+x_{1}=x_{1} - C_{2}$
$........$
所有式子带入之后其实就是求$|x_{1}|+|x_{1} - C_{1}|+|x_{1} - C_{2}|+ ....+|x_{1} - C_{n-1}|$的最小值，这个其实就是在一个坐标轴上找一点$x$使得$x$和$0,C_{1},C_{2}..C_{n-1}$的距离之和最短，这个是求中位数，对这些数排序一下$x$取中位数就好了

现在看有$n$个人隔$k$个人才能给硬币，其实这个可以看做$gcd(n,k+1)$个圈，然后他们之间相邻的两个人可以交换硬币，第$1$个人依次和$1 + (k+1),1 + 2(k+1)...$形成一个圈，剩余的人也是这样，这样就是处理$gcd(n,k+1)$个圈就是了，最终结果全部相加就是答案，注意一些特殊情况，$k = n-1$或者$k=n$其实是不能交换的状态，这个特判一下就好了

#include<bits/stdc++.h>
typedef long long ll;
const ll maxn = 1e6 + 5;
const ll INF = 1e18;
using namespace std;

ll res[maxn], C[maxn];
ll solve(ll n, ll fnl) {
    C[0] = 0;
    for(ll i = 1; i < n; i++) C[i] = C[i - 1] + res[i] - fnl;
    sort(C, C + n);
    ll x = C[n / 2], ans = 0;
    for(ll i = 0; i < n; i++) ans += abs(x - C[i]);
    return ans;
}

ll a[maxn], k, n;
ll vis[maxn], m;

int main() {
    while(scanf("%lld %lld", &n, &k) != EOF) {
        memset(vis, 0, sizeof vis); k++;
        ll s = 0;
        for(ll i = 0; i < n; i++) { scanf("%lld", &a[i]); s += a[i]; }
        if(k >= n) {
            ll fl = 1;
            for(ll i = 0; i < n; i++) if(a[i] != s / n) fl = 0;
            if(fl) cout << "0" << endl;
            else cout << "gg" << endl;
            continue;
        }
        ll fnl = -1, flag = 1, ans = 0;
        for(ll i = 0; i < n; i++) if(!vis[i]) {
            ll num = 1, sum = 0, tot = 0;
            for(ll j = i; !vis[j]; j = (j + k) % n) {
                vis[j] = 1; res[num++] = a[j];
                sum += a[j]; tot++;
            }
            if(sum % tot) { flag = 0; break; }
            if(fnl == -1) fnl = sum / tot;
            if(sum / tot != fnl) { flag = 0; break; }
            ans += solve(num - 1, fnl);
        }
        if(!flag) cout << "gg" << endl;
        else  cout << ans << endl;
    }
    return 0;
}