题意:求所给公式ans的最大值。 n1 n2 n1 n2
解题思路:由于n1+n2=N,n1*v1+n2*v2=S,S是所有数的和。ans = ∑(Ai-V2)^2+∑(Bi-V1)^2=∑Ai^2+∑Bi^2-2*N*v1*v2+
i=1 i=1 i=1 i=1
n1*v2*v2+n2*v1*v1,对于特定的n1,n2,将v2全部由v1替换,可以将上式化简ans =C1*v1*v1-C2*v1+C3的形式,其中C1, C2, C3 是正常数。所以ans 是开口向上的二次函数,最大值一定是在v1 的两个端点取得,所以对于特定的n1, n2,当v1 取得最大值或者最小值时ans 有最大值。根据上面的证明,只需要将所有的数排序,最大值的两个集合一定是从中间某个位置切开分成的集。详见代码:
Code:
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int Arrays[1010];
int main()
{
int T,N;
scanf("%d",&T);
for(int k = 1; k<=T; k++)
{
scanf("%d",&N);
double Sum1 = 0,Sum2 = 0;
for(int i = 1; i<=N; i++)
{
scanf("%d",&Arrays[i]);
Sum1 +=(Arrays[i]*Arrays[i]);
Sum2 +=Arrays[i];
}
sort(Arrays+1,Arrays+N+1);
double temp = 0,ans =-1e9;
for(int i = 1; i<=N-1; i++)
{
temp +=Arrays[i];
double v1 = temp/i;
double v2 = (Sum2-temp)/(N-i);
ans = max(ans,i*v2*v2+(N-i)*v1*v1-2*N*v1*v2);
}
printf("Case #%d: %.6f\n", k,ans+Sum1);
}
return 0;
}