集训笔记：ECNA 2023-2024 B-B Road Band题解

题目类型： 线性dp，数学

题意：

​ 给定两条平行轴上的点，两条轴的距离是$s$。第一条轴上给定$m$个点，第二条轴上给定$n$个点，点坐标$x$。在距两条轴距离相等且平行的线上，标记$k$个基站，

​ 求$n+m$个点到最近的基站的距离平方和。要求最小，并输出这个值，误差不超过$1e^{-5}$

​ 数据范围：$m(m\le 1000)$，$n(n\le 1000)$，$x(0\le x\le 1000)$，$k(k\le 100)$

模型

​ 根据题意，基站一定在中间，所以在两岸的点是等价的。

​ 因此可以把这个题的两岸的点转移到一个岸上，即线上修建$k$个点，轴上$n+m$个点，到线的距离为$\frac{s}{2}$，求出每个点到最近基站的距离平方的和最小。

​ 那么每个修建的基站对应的是一段连续的点的区间，枚举每一个修建基站对应了哪一段点。使用线性dp模型求解此题。

最优子结构

​ 设点的序列X =< $x_1,x_2,...,x_{n+m}$ >，基站的序列是Y = <$y_1,y_2,...,y_k$>。因此，$f_{i,j}$表示前i个点建j个基站的最小距离平方和，满足下列条件

1.如果只有一个基站（j = 1），则点序列（$x_1-x_{n+m}$）全部连接一个基站。

2.如果有多个基站并且点数大于基站数（j > 1 && i > j），则不需要再建立新的基站，$f_{i,j}$ = $f_{i,i}$

3.如果有多个基站并且点数不大于基站数（j > 1 && i <= j），则枚举点序列X1 =<$x_1,x_2,...x_j$>$k<i$ ，序列X2 =<$x_{k+1},x_{k+2},...x_i$> 序列X1用j - 1个基站，序列X2用剩下一个基站。

点数\基站数量	1	2	…	k
1	$f_{1,1}$	$f_{1,2}$	…	$f_{1,k}$
2	$f_{2,1}$	$f_{2,2}$	…	$f_{2,k}$
…	…	…	…	…
n + m	$f_{n+m,1}$	$f_{n+m,2}$	…	$f_{n+m,k}$

递归解

​ $dp[i][j]$ 表从第$1$个点到第$i$个点，建$j$个基站的最小距离平方和。
$$dp[i][j]=\{\begin{array}{l}ask(1,i),j=1\\ dp[g][j-1]+ask(g+1,i),1<j\le i,1\le g<i\\ dp[i][j-1],j>i\end{array}$$
其中，$ask(st,ed)$ 表示从$st$点到$ed$点，建一个基站，最小距离平方和。

时间复杂度

​ 子问题状态n+m * k 个，状态转移n+m次，复杂度为O((n+m) * k * (n+m))

实现

ask函数

​ 由勾股定理得，点到基站的距离
$$\begin{array}{rl}y& =\sum \left({\left(a_i-x\right)}^2+{\left(\frac{s}{2}\right)}^2\right)\\ & =\sum a_i^2-2\sum a_i\ast x+\sum x^2+\sum {\left(\frac{s}{2}\right)}^2\end{array}$$

​ 结论：一段点建立基站的位置是这几个点的坐标平均值。

​ 证明方法一：

​ 对k个点，有
$$y_k=\sum _{i=1}^k{a}_i^2-2\sum _{i=1}^k{a}_i\ast x+\sum x^2+k\ast (\frac{s}{2}{)}^2$$
​ 令$a_{avg}=\frac{{\sum }_{i=1}^k{a}_i}{k}$

$$y_k=k(a_{avg}^2-2\ast a_{avg}\ast x+x^2)=k(a_{avg}-x{)}^2$$
显然$y_k$是一个开口向上的二次函数。最小值点$x=a_{avg}$

​ 证明方法二：

​ 对k个点，有
$$f_k=\sum _{i=1}^{n+m}(a_i-x{)}^2+(\frac{s}{2}{)}^2$$

$$f^{\prime }=2(a_1-x)+2(a_2-x)+...+2(a_i-x)$$

​ 令$f^{\prime }=0$，则
$$x_0=\frac{{\sum }_{j=1}^i{a}_j}{i}$$
并且，在$x_0$左边$f^{\prime }<0$递减，$x_0$右边$f^{\prime }>0$，因此$x_0$是极小值点。

​ 对于y我们只要维护 $\sum a_i^2$ ,$\sum a_i$ ,${\left(\frac{s}{2}\right)}^2$ ，即前缀和与前缀平方和。

完整代码

#include <bits/stdc++.h>
using namespace std;

void solve() {
    int m, n, k;
    double s;
    cin >> m >> n >> k >> s;

    vector<double> pre(n + m + 10, 0), pre2(n + m + 10, 0);
    vector<double> a(n + m + 10);

    for (int i = 1; i <= n + m; i++) {
        cin >> a[i];
    }
    sort(a.begin(), a.begin() + n + m + 1);
    for (int i = 1; i <= n + m; i++) {
        pre[i] = pre[i - 1] + a[i];
        pre2[i] = pre2[i - 1] + a[i] * a[i];
    }

    vector<vector<double>> dp(n + m + 10, vector<double>(k + 10));
    auto ask =[&] (int st, int ed) {
        int num = ed - st + 1;
        double add1 = pre[ed] - pre[st - 1];
        double add2 = pre2[ed] - pre2[st - 1];
        double add3 = (s / 2) * (s / 2) * num;

        double x = add1 / num;
        double sub = 2 * x * add1;
        double res = num * x * x - sub + add2 + add3;

        return res;
    };

    for (int i = 1; i <= n + m; i++) {
        dp[i][1] = ask(1, i);
        for (int j = 2; j <= k; j++) {
            dp[i][j] = ask(1, i);
            for (int g = 1; g < i; g++) {
                dp[i][j] = min(dp[i][j], dp[g][j - 1] + ask(g + 1, i));
            }
        }
    }

    cout << fixed << setprecision(9);
    cout << dp[n + m][k] << "\n";
}

signed main() {
    ios::sync_with_stdio(0);
    cin.tie(0), cout.tie(0);

    int T = 1;
    // cin >> T;
    while(T--) {
        solve();
    }
    return 0;
}