【题解】codeforces792E Colored Balls

题目链接

题意：给定n个正整数a[1]，...，a[n]，现在要将每个正整数拆成若干个小正整数的和，使得任意2个小正整数的差不超过1（即取一个正整数k，将a[1]，...，a[n]拆成若干个k 与 若干个k+1）。求使得小正整数总数最少的方案，输出该方案下的小正整数总数。

分析：假如我们知道最优的情况下小正整数的值应为k和k+1，那么我们很容易在O(n)的时间复杂度内求出这种情况下拆分得到的最少小正整数总数。

        假设在最优的方案下小正整数为k和k+1，并假设将a[1]拆成x个k和y个k+1，那么有x*k+y*(k+1)=a[1]，从而(x+y)*k<=a[1]<=(x+y)*(k+1)。从而k<=a[1]或x+y<=a[1]。

        若k<=a[1]，那么直接枚举k的值就行了。

        若x+y<=a[1]，那么就枚举x+y的值。由于a[1]/(x+y)-1<=k<=a[1]/(x+y)，所以对每个x+y，至多有2个k的值是合法的，所以再根据x+y的值求k的值即可。

        时间复杂度为O(nsqrt(a[1]))。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn=500+10;
int n;
LL a[maxn],ans=-1;
void calc(int k)
{
LL temp=0;
for (int i=1;i<=n;i++)
{
int x=a[i]/k,y=a[i]%k;
if (x<y) return ;
x-=y;
temp+=(x+y-x/(k+1));
}
if (ans==-1||ans>temp) ans=temp;
}
int main()
{
cin>>n;
for (int i=1;i<=n;i++) scanf("%I64d",&a[i]);
for (int k=1;(LL)k*k<=a[1];k++)
{
calc(k);calc(a[1]/k);
if (a[1]%k==0&&a[1]>k) calc(a[1]/k-1);
}
cout<<ans;
return 0;
}