hdu5073 贪心 + 公式拆解

分析:

我来就大胆猜了一个很有道理的结论,一定是连续的 nk 个不变,然后扫一遍计算。很巧的一个贪心。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <stack>
using namespace std;
typedef pair<int, int> pii;
typedef long long ll;
typedef unsigned long long ull;
#define xx first
#define yy second
#define pr(x) cout << #x << " " << x << " "
#define prln(x) cout << #x << " " << x << endl
template<class T> inline T lowbit(T x) { return x & (-x); }

const int MAXN = 5e4 + 5;
double sum[MAXN], sqr_sum[MAXN];
int d[MAXN];

int T;
int n, k;
int main(void) {
#ifdef LOCAL 
    freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
#endif
    scanf("%d", &T);
    while(T--) {
        scanf("%d%d", &n, &k);
        for (int i = 1; i <= n; i++) {
            scanf("%d", &d[i]);
        }
        if(n == k) {
            puts("0");
            continue;
        }
        k = n - k;
        sort(d + 1, d + n + 1);
        for (int i = 1; i <= n; i++) {
            sum[i] = sum[i-1] + d[i];
            sqr_sum[i] = sqr_sum[i-1] + double(d[i]) * d[i];
        }
        double ans = 1e18;
        for (int i = k; i <= n; i++) {
            double o = (sum[i] - sum[i-k]) / k;
            ans = min(ans, sqr_sum[i] - sqr_sum[i-k] -
                    2*(sum[i] - sum[i-k])*o + k * o * o);
        }
        printf("%.15f\n", ans);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值