UVALive 7049 Galaxy

#include<bits/stdc++.h>
#define cl(a,b) memset(a,b,sizeof(a))
using namespace std;

const int maxn=5e4+10;

int star[maxn],d[maxn];
double d2[maxn];

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,k;
        scanf("%d%d",&n,&k);
        cl(star,0),cl(d,0),cl(d2,0);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&star[i]);
        }
        sort(star+1,star+n+1);
        for(int i=1;i<=n;i++)
        {
            d[i]=d[i-1]+star[i];
            d2[i]=d2[i-1]+1.0*star[i]*star[i];
        }
        int l,r;
        double ans=1e18,x;
        if(n==k)
        {
            printf("0
");
            continue;
        }
        for(int i=1;i<=k+1;i++)
        {
            l=i,r=n-(k-i+1);
            int tmpd=d[r]-d[l-1];
            x=1.0*tmpd/(r-l+1);
            ans=min(ans,d2[r]-d2[l-1]-2*x*tmpd+x*x*(r-l+1));
        }
        printf("%f
",ans);
    }
    return 0;
}/*

2
3 2
-1 0 1
4 2
-2 -1 1 2

*/