题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=5073
题意:给出几颗行星的位置,你可以移动其中若干颗行星的位置,求移动后每个行星到整个星系中心的距离的平方和的最小值,行星的个数为n,可移动的个数为k
思路:
贪心思路 1. 将k颗行星都移到其他n-k颗行星的中心,则这k颗的距离都为零,原问题简化成了从n颗中取出n-k颗,使其平方和最小.
贪心思路 2. 根据观察发现了,中心的坐标其实就是所有坐标的平均值,所以原问题其实就是求一个数列的方差的最小值,由于方差描述了一个串数的波动情况,所以所取的n-k个数一定要是连续的,才能使其方差最小。
数学推导: 设n个数为x1,x2,x3,,,,,,,xn,其平均值为xx,所以有
(x1-xx)^2+(x2-xx)^2+(x3-xx)^2+,,,,,+(xn-xx)^2=(x1^2+x2^2+,,,,,,+xn^2)-2*xx*(x1+x2+...+xn)+n*xx*xx
所以我们维护一个和,再维护一个平方和就能快速的计算出一段序列的方差。 其复杂度为O(n) 排序复杂度为O(nlogn)。
要注意n==k的情况,这个时候会除零。
code:
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn=55000;
double a[maxn];
int main()
{
int T,n,k,cnt;
double sum1,sum2,ans,xx;
scanf("%d",&T);
while(T--){
scanf("%d%d",&n,&k);
for(int i=0;i<n;i++) scanf("%lf",&a[i]);
sort(a,a+n);
xx=ans=sum1=sum2=0;
cnt=n-k;
if(!cnt){
printf("0\n");
continue;
}
for(int i=0;i<cnt;i++){
sum1+=a[i];
sum2+=a[i]*a[i];
}
xx=sum1/cnt;
ans=sum2-2*xx*(sum1)+cnt*xx*xx;
for(int i=cnt;i<n;i++){
sum1-=a[i-cnt];
sum2-=a[i-cnt]*a[i-cnt];
sum1+=a[i];
sum2+=a[i]*a[i];
xx=sum1/cnt;
ans=min(ans,sum2-2*xx*sum1+cnt*xx*xx);
}
printf("%f\n",ans);
}
return 0;
}