hdu5073-思维&方差-Galaxy

http://acm.hdu.edu.cn/showproblem.php?pid=5073
背景知识很强qwq
给定一些星星，你可以删除k个，
每一个星星
s=d*d.(d为每个星星距离质心的距离。)，问你如何使sigma（s）最小。
展开公式得
(x1-zx)(x1-zx)+(x2-zx)(x2-zx)…
x1*x1+x2*x2+…. +(m-k)*zx*zx-2*sigma(x)*zx>.
并且发现这个结果和 区间的方差呈正相关，即越密集，越小。
于是枚举 连续的区间就行。。

#include <bits/stdc++.h>
using namespace std;
/*  思维
  给定你n个点，要求你去掉k个点，使其留下的点
  方差最小。

*/
const int maxn=5e4+1000;
double  a[maxn];
int main()
{   int t,m,k;
    scanf("%d",&t);
    while(t--){
         scanf("%d%d",&m,&k);
         for(int i=1;i<=m;i++){
             scanf("%lf",&a[i]);
         }
         sort(a+1,a+m+1);
         if(m==k){
            puts("0");
            continue;
         }
         double sum1=0;
         double sum2=0;
         for(int i=1;i<=m-k;i++){
            sum1+=1.0*a[i];
            sum2+=1.0*(a[i]*a[i]);
         }
         double  zx=sum1/(m-k);
         double min1=sum2-2*zx*sum1+(m-k)*zx*zx;
         for(int i=m-k+1;i<=m;i++){
            sum1-=a[i-m+k];
            sum2-=a[i-m+k]*a[i-m+k];
            sum1+=a[i];
            sum2+=a[i]*a[i];
            zx=sum1/(m-k);
            min1=min(min1,sum2-2*zx*sum1+(m-k)*zx*zx);
         }
         printf("%.10f\n",min1);



    }



    return 0;
}