石子游戏（贪心+思维）

石子游戏

时间限制: 1 Sec   内存限制: 128 MB
提交: 69   解决: 16
[ 提交][ 状态][ 讨论版][命题人: admin]

题目描述

在Bob学会怎样玩Nim Game之后，他打算尝试另一款看起来更为简单的石子游戏
这个游戏是这样子玩的：一共有一个玩家，且一开始有N堆石头，第i堆石头有ai个石子。玩家每次只能移动一个石子从一堆到另一堆。在每次移动结束后，如果存在一个整数x(x>1)满足任意一堆的当前石子数bi都是x的倍数，那么游戏结束。现在你需要帮助Bob计算出为了结束这个无聊的游戏，他最少需要移动的次数。特别的， 0是任何正整数的倍数。

输入

第一行一个整数N，表示石子的堆数
第二行N个整数，表示每堆石子的数量

输出

一行一个整数，即最少的移动次数。如果一开始就满足游戏结束的条件，请输出0

样例输入

5

1 2 3 4 5

样例输出

2

提示

从第1堆移动一个到第5堆，从第4堆移动一个到第2堆
得到：0 3 3 3 6
满足都是3的倍数
数据规模
对于30%的数据，1<=N<=100
对于60%的数据，1<=N<=5000

对于100%的数据，1<=N<=100,000，1<=ai<=100,000

由于最后得到的数都是x的倍数，所以x必然是素数，若保证最后得到的数是x的倍数，则需要所有数的和是x的倍数。求素数O(nlogn)，然后枚举素数，排除掉不等于0的状况，时间复杂度进一步优化。然后就是求每个数跟x的模除的余数，算出总的余数可以构成多少个x，余数从大到小开始凑，时间复杂度进一步减少，维护一个余数和，当余数和整除x等于前面得到的数的时候，就可以跳出循环。总的时间复杂度O(k*nlogn)，k是满足要求的素数的个数。

代码实现：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<cstdio>
#define ll long long
#define mset(a,x) memset(a,x,sizeof(a))

using namespace std;
const double PI=acos(-1);
const int inf=0x3f3f3f3f;
const double eps=1e-12;
const int maxn=1e6+5;
const int mod=1e9+7;
int dir[4][2]={0,1,1,0,0,-1,-1,0};
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
ll lcm(ll a,ll b){return a/gcd(a,b)*b;}
ll inv(ll b){if(b==1)return 1; return (mod-mod/b)*inv(mod%b)%mod;}
ll fpow(ll n,ll k){ll r=1;for(;k;k>>=1){if(k&1)r=r*n%mod;n=n*n%mod;}return r;}
ll Fpow(ll n,ll k){ll r=1;for(;k;k>>=1){if(k&1)r=r*n;n=n*n;}return r;}
ll upd(ll x,ll v){x=x+v>=mod?x+v-mod:x+v;return x;}
int s[maxn],prim[maxn],vis[maxn];
int m[maxn];

int main()
{
	ll n,i,j,k;
	k=0;
	for(i=2;i<=100000;i++)
	{
		if(!vis[i])
		{
			prim[k++]=i;
			for(j=i*i;j<=100000;j+=i)
			vis[j]=1;
		}
	}
	ll sum=0;
	scanf("%lld",&n);
	for(i=0;i<n;i++)
	{
		scanf("%d",&s[i]);
		sum+=s[i];
	}
	ll minn=1e18;
	for(i=0;i<k;i++)
	{
		if(sum%prim[i])
		continue;
		ll t=0;
		for(j=0;j<n;j++)
		{
			m[j]=s[j]%prim[i];
			t+=m[j];
		}
		t=t/prim[i];
		sort(m,m+n);
		ll sum_1=0,sum_2=0,sum_3=0;
		for(j=n-1;j>=0;j--)
		{
			sum_1+=(m[j]+prim[i]-m[j]%prim[i]);
			sum_2=sum_1/prim[i];
			sum_3+=prim[i]-m[j]%prim[i];
			if(sum_2==t)
			{
				minn=min(minn,sum_3);
				break;
			}
		}
	}
	printf("%lld\n",minn);
}