hdu 5073 Galaxy 2014 鞍山区域赛 数学

题目

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5073

题目来源：gdut钢霸给的

简要题意：数轴有$n$个点，可以移动其中$k$个，要取数轴一点令它到给定点的距离平方和最小。

数据范围：$T \leqslant 10;\quad 1 \leqslant n \leqslant 50000;\quad 0 \leqslant k \leqslant n;\quad |x_i| \leqslant 50000$

题解

移动的$k$个点一定是移动到你设定的那个点最好，因为这样是$0$。

这样只需要考虑剩余$n-k$个点的情况。

不妨先固定下选定的点，为方便直接记数量为$m=n-k$，要求下面这东西。

$\sum\limits_{i=1}^{m}{(p-x_i)^2} =\sum\limits_{i=1}^{m}{(p^2-2px_i+x_i^2)} =mp^2-2mp\overline{X}+\sum\limits_{i=1}^{m}{x_i^2}$

由于要求的是最小值，可以知道对称轴取抛物线对称轴$p=\overline{X}$，即平均数。

结果可以化简为$\sum\limits_{i=1}^{m}{x_i^2}-m\overline{X}^2$，其实这玩意就是概率论里的二阶中心距。

对$x_i$排序，$m$个点应该是连成一块的，使用反证法可以进行证明。

于是最后只需要排序然后枚举就行了。

实现

为了快速得到$\sum\limits_{i=1}^{m}{x_i^2}-m\overline{X}^2$需要进行$x_i,x_i^2$的前缀和预处理。

注意判断边界上的情况，$n=k$的时候要特判。

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>

#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef vector<int> VI;
typedef pair<int,int> PII;
LL powmod(LL a,LL b, LL MOD) {LL res=1;a%=MOD;for(;b;b>>=1){if(b&1)res=res*a%MOD;a=a*a%MOD;}return res;}
// head
const int N = 1E5+5;
int a[N];
int sum[N];
LL sqsum[N];

int main()
{
    int t, n, k;
    scanf("%d", &t);
    while (t--) {
        scanf("%d%d", &n, &k);
        for (int i = 1; i <= n; i++) {
            scanf("%d", a+i);
        }
        if (k == n) {
            puts("0");
            continue;
        }
        sort(a+1, a+n+1);
        for (int i = 1; i <= n; i++) {
            sum[i] = a[i]+sum[i-1];
            sqsum[i] = sqsum[i-1]+LL(a[i])*a[i];
        }
        double ans = 1e18;
        int len = n-k;
        for (int i = 1; i <= k+1; i++) {
            int be = i, en = i+len-1;
            double avg = double(sum[en]-sum[be-1])/(len);
            ans = min(ans, sqsum[en]-sqsum[be-1]-avg*avg*len);
        }
        printf("%.12f\n", ans);
    }
    return 0;
}