题目:Be Together
题意很明确就是把一个数组中每个数都变成相等的,且变化的数值尽量小。
解法一:看到这题我第一反应想到是平均数,这样每个数的变化相对来说会比较小.但是计算平均数又有个问题了,设sum为数组的总和,如果sum%n==0那好说,可是sum%n!=0呢,到底向上取整还是向下取整呢.这个取决数组了.所以分清况计算一下就行了.(之前想当然的向下取整WA了几发)
#include<bits/stdc++.h>
#define ll long long
const long double pi=acos(-1);
const int N=2e5+10;
using namespace std;
int arr[N];
int main()
{
ll n;
ll sum1=0,sum=0;
cin>>n;
for(int i=0;i<n;i++)
{
cin>>arr[i];
sum+=arr[i];
}
ll tmp=sum/n;
for(int i=0;i<n;i++)
{
ll k=(arr[i]-tmp);
sum1+=k*k;
} //这个情况是sum%n==0和sum%n!=0向下取整的情况.
tmp++;
ll sum2=0;
for(int i=0;i<n;i++)
{
ll k=(arr[i]-tmp);
sum2+=k*k;
}
sum1=min(sum1,sum2);
cout<<sum1<<"\n";
return 0;
}
解法二:这个题很明显的特点就是ai都很小,-100<=ai<=100;最大才100,整个区间不过200.既然ai两端最值都确定了,那么最优解区间也就确定了.遍历一遍-100~200就可以了.
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn=100+10;
const int inf=1<<30;
int a[maxn];
int main()
{
int n;
scanf("%d",&n);
int sum=0;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]),sum+=a[i];
int ans1=0,ans=inf;
for(int j=-100;j<=100;j++)
{
ans1=0;
for(int i=1;i<=n;i++)
{
int k=a[i]-j;
ans1+=k*k;
}
ans=min(ans,ans1);
}
cout<<ans<<"\n";
}