首页 > 代码库 > hdu 5073 有坑+方差贪心

hdu 5073 有坑+方差贪心

http://acm.hdu.edu.cn/showproblem.php?pid=5073

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


根据方差的性质(越密越小)得知贪心策略就是,n-k序列一定连续,那么随即求连续n-k序列里面方差最小的那个

设连续长度为k

avg = sum/k;

有 求和(pi - avg)^2 = (求和pi^2)   -  sum^2/k;

然后贪心模拟一遍即可

输入序列式无序的,样例坑到死,不sort或者sum^2/k精度问题WA到死


#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <queue>
#include <stack>
#include <iostream>
#include <algorithm>
using namespace std;
#define RD(x) scanf("%I64d",&x)
#define RD2(x,y) scanf("%d%d",&x,&y)
#define RD3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define clr0(x) memset(x,0,sizeof(x))
typedef long long LL;
const int maxn = 5e4+5;
const double INF = 1e20;
int n,k;
LL p[maxn],sum[maxn],sq[maxn];
int main() {
    int _;RD(_);while(_--){
        RD2(n,k);
        p[0] = sum[0] = sq[0] = 0;
        for(int i = 1;i <= n;++i)
            RD(p[i]);
        double ans = INF;
        k = n - k;
        if(k == 0 || k == 1){
            puts("0");
            continue;
        }
        sort(p+1,p+n+1);
        for(int i = 1;i <= n;++i)
            sum[i] = p[i] + sum[i-1],sq[i] = p[i]*p[i] + sq[i-1];
        for(int i = k;i <= n;++i)
            ans = min(ans,(double)(sq[i] - sq[i - k])*(double)k - (double)(sum[i] - sum[i-k])*(double)(sum[i] - sum[i-k]));
        printf("%.10lf\n",ans/(double)k);
    }
    return 0;
}


hdu 5073 有坑+方差贪心