HDU 6237 A Simple Stone Game

这是一个非常interesting的题目

HDU6237传送门
题目大意就不说了。
日常练习的时候刷2017哈尔滨CCPC的题。
果断一题崩盘。
练习后参考了别人的思路写的这道题。

---

我们在练习的时候想法是：
1.在游戏开始时先判断每个数字的GCD，如果大于1则答案为0。接下来只考虑GCD==1的情况；
2.如果所有堆的石子总数和为偶数，且GCD为1，那么应该把所有的奇数堆消灭掉变成偶数堆（这个推论可能是错的）;
3.目标状态为，所有堆石子的数量都为X的倍数，那么可以得到一个整体局面。设石子总数为sum，第i堆石子的个数为 ki*X，则有sum=（k1+k2+k3+…+kn）*x。所以得到一个重要结论是X必然为sum的一个因子。但是因子总数太多，不知道该如何枚举。

---

其实思路已经接近了一些，但是无奈长时间不做数论题思路不能继续。
看完题解恍然大悟。
在sum的众多因子里只需要枚举质因子（个数不会超过log2（sum））即可。如果质数prime考虑过，那么k*prime必然也已经考虑过，所以没有必要继续枚举跟prime的倍数。

---

当我们枚举sum的一个质因子prime并开始验证时，先把各个堆的个数取余。利用贪心的想法，优先将有较多石子的堆凑到prime个，在小堆石子中取石子放到较大的堆中。计量操作数比较有技巧，当考虑一个有n个石子的堆时，我们需要操作prime-n次把它凑到prime个。操作后，有prime个石子被考虑完毕。所以所有的石子都被考虑过，流程就结束了。这里在自己写的时候试图用双指针去写，结果wa了。:(

---

综上所述可以得出答案：
1.先特判出不需要操作的情况。
2.求出总数sum并得出sum所有的质因子。
3.对于每一个质因子进行验证，看操作到目标局面最少的操作数是多少。输出最小值。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

const int maxn=1e5+20;

long long arr[maxn];
vector <long long> prime;
vector <long long> num;
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n;
        scanf("%d",&n);
        long long sum=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&arr[i]);
            sum+=arr[i];
        }

        prime.clear();

        for(long long i=2;i*i<=sum;i++)
        {
            if(sum%i==0)
            {
                prime.push_back(i);
                while(sum%i==0)
                    sum/=i;
            }
        }
        if(sum!=1)  prime.push_back(sum);

        sort(prime.begin(),prime.end());

        long long ans=1LL<<62;
        for(int i=0;i<prime.size();i++)
        {
            num.clear();
            long long x=prime[i];
            long long cur=0;
            for(int j=1;j<=n;j++)
            {
                long long memo=arr[j]%x;
                if(memo)
                {
                    num.push_back(memo);
                    cur+=memo;
                }
            }
            sort(num.begin(),num.end());
            long long check=0;
            for(int j=num.size()-1;j>=0;j--)
            {
//                cout<<x<<" "<<prime[j]<<" "<<"----"<<x-prime[j]<<endl;
                check+=x-num[j];
                cur-=x;
                if(cur==0)  break;
            }
            ans=min(ans,check);
        }
        printf("%lld\n",ans);
    }
    return 0;
}