HDU-6237 A Simple Stone Game

给出一个数组a，问需要移动多少石子才能让所有a[i]%x==0，（x>1），a[i]==0时满足a[i]余任意x等于0

首先可以确定所有a[i]的和sum%x==0，那么我们就缩小了x的范围，就是sum的因子

并且如果x不是质数，那么x就可以用它的因子来替代，结果是一样的，所以我们只需要考虑x是质数的情况

在给定x后，最少需要移动的石子个数怎么计算

先用a[i]%x得到数组b[i]，统计b[i]的和b_sum，统计出来的余数需要重新组成b_sum/x个石子堆

我们选b[i]中前b_sum/x个最大石子堆作为放置点，其它所有的石子就都是需要移动的

所以我们就在2~sqrt(sum)+1的范围内遍历所有质数作为x，对于每个x计算最少需要移动的石子数

别忘了还有将sum本身作为x的情况，最后每个x算出来的个数取最小值就是答案

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<stack>
using namespace std;
typedef long long ll;
const int N=1e5+10;
int prime[N];
void get_prime() //要用快一些的素数打表，否则可能会超时
{
	memset(prime,0,sizeof(prime));
	for(int i=2;i<=N;i++)
	{
		if(!prime[i]) prime[++prime[0]]=i;
		for(int j=1;j<=prime[0]&&prime[j]<=N/i;j++)
		{
			prime[prime[j]*i]=1;
			if(i%prime[j]==0) break;
		}
	}
}
ll a[N],sum,b[N];
int main()
{
	int n,T;
	get_prime();
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		sum=0;
		for(int i=0;i<n;i++)
		{
			scanf("%lld",&a[i]);
			sum+=a[i];
		}
		ll x,l=sqrt(sum)+1,ans=sum;
		for(int i=1;i<=prime[0]&&prime[i]<=l;i++)
		{
			x=prime[i];
			if(sum%x!=0) continue;
			int cnt=0;
			ll num=0;
			for(int j=0;j<n;j++)
			{
				ll m=a[j]%x;
				if(m) b[cnt++]=m,num+=m;
			}
			sort(b,b+cnt);
			ll t=0,p=num/x;
			for(int j=0;j<cnt-p;j++)
				t+=b[j];
			ans=min(ans,t);
		}
		int cnt=0;
		for(int i=0;i<n;i++)
		{
			ll m=a[i]%sum;
			if(m) b[cnt++]=m;
		}
		sort(b,b+cnt);
		ll t=0;
		for(int i=0;i<cnt-1;i++)
			t+=b[i];
		ans=min(ans,t);
		printf("%lld\n",ans);
	}
	return 0;
}