hdu 2014鞍山赛区 5073 Galaxy

题意：就是给你 n 个数，代表n个星球的位置，每一个星球的重量都为 1 ！
开始的时候每一个星球都绕着质心转动，那么质心的位置就是所有的星球的位置之和 / 星球的个数
现在让你移动 k 个星球到任意位置（多个星球可以在同一个位置并且所有的星球在同一直线上）
移动之后那么它们质心的位置就可能发生变化，求 I = sum(di^2) di （表示第i个星球到达质心的距离）最小！

设d为n-k个星球的质心位置，如果I值最小，那么移动的k个星球一定都放在另外n-k个星球的质心上，
并且这n-k个星球一定是连续的！越密集方差越小嘛.....
x1, x2, x3, x4,....x(n-k)表示余下n-k个星球的位置
思路：I = sum(di^2) = (x1-d)^2 + (x2-d)^2 + (x3-d)^2 ....
= sum(xi^2) + (n-k)*d*d - 2*d*sum(xi);

 #include<iostream>

 #include<cstring>

 #include<cstdio>

 #include<algorithm>

 #define N 50050

 using namespace std;

 double num[N];

 double s1[N], s2[N];

 int main(){

     int n, t, k;

     scanf("%d", &t);

     while(t--){

         scanf("%d%d", &n, &k);

         for(int i=; i<=n; ++i)

             scanf("%lf", &num[i]);

         sort(num+, num+n+);//对星球的位置排一下序

         for(int i=; i<=n; ++i)//分别计算前缀num[i] 的和 以及 num[i]^2的和

             s1[i] = s1[i-] + num[i], s2[i] = s2[i-] + num[i]*num[i];

         int m = n-k;

         double ans = 1000000000000000000.0;//ans要足够大.... 最好不用long long，可能会超....

         for(int i=; m && i+m-<=n; ++i){

             int j = i+m-;

             double d = (s1[j] - s1[i-])/m;

             double tmp = s2[j] - s2[i-] - *d*(s1[j] - s1[i-]) + m*d*d;

             if(ans > tmp) ans = tmp;

         }

         if(n == k) ans = 0.0;

         printf("%.9lf\n", ans);

     }

     return ;

 }

巴特西

hdu 2014鞍山赛区 5073 Galaxy

最新文章

热门文章