leetcode 1478. 安排邮筒

题目描述

给你一个房屋数组houses 和一个整数 k ，其中 houses[i] 是第 i 栋房子在一条街上的位置，现需要在这条街上安排 k 个邮筒。

请你返回每栋房子与离它最近的邮筒之间的距离的 最小 总和。

答案保证在 32 位有符号整数范围以内。

示例 1：

输入：houses = [1,4,8,10,20], k = 3
输出：5
解释：将邮筒分别安放在位置 3， 9 和 20 处。
每个房子到最近邮筒的距离和为 |3-1| + |4-3| + |9-8| + |10-9| + |20-20| = 5 。
示例 2：

输入：houses = [2,3,5,12,18], k = 2
输出：9
解释：将邮筒分别安放在位置 3 和 14 处。
每个房子到最近邮筒距离和为 |2-3| + |3-3| + |5-3| + |12-14| + |18-14| = 9 。
示例 3：

输入：houses = [7,4,6,1], k = 1
输出：8
示例 4：

输入：houses = [3,6,14,10], k = 4
输出：0
提示：

n == houses.length
1 <= n <= 100
1 <= houses[i] <= 10⁴
1 <= k <= n
数组 houses 中的整数互不相同。

分析

本题的前置题AcWing 104 货仓选址，也就是说，做本题的前置知识就是，如果在数轴上有若干个点，选一个坐标，使其离所有点距离之和最小，那么如果数轴上有奇数个点，选定的坐标就是最中间的点；如果有偶数个点，选定的坐标就是最中间两个点之间的任意位置。理论证明见前面货仓选址的题解。

既然选一个点建货仓找中位数就可以了，那么选k个点呢？可以先选好k - 1个点，再来考虑最后一个货仓的选址。准确的说，是只需要枚举最后一个货仓负责的房子，比如10个房子建3个货仓，那么最后一个货仓可以负责1到8个房子。状态表示：f[i][j]表示前i个房子建j个货仓的最小距离和，那么f[i][j] = min(f[k][j-1] + c),其中k从j - 1到i - 1，c表示最后一个货仓到剩下房子的最小距离之和。

求任意区间建一个货仓的最小距离之和可以初始化一下，一个方法是使用前缀和，比如x1 x2 x3 x4四个房子，货仓建在x2和x3中间，距离之和就是x3 - x2 + x4 - x1 = （x3 + x4） - （x1 + x2）；如果是奇数个房子，比如x1到x5，货仓建在x3，距离之和与四个房子一样。另一个方法是DP求解，用c[i][j]表示i到j的房子建一个货仓的最小距离之和，则状态转移方程为c[i][j] = c[i+1][j-1] + houses[j] - houses[i]。对应上面的例子就是c[1][4] = c[2][3] + x4 - x1。初始化出任意一段区间内建一个货仓的距离之和后，就可以在O（n²）的时间内求出本题的解了。

有一点需要注意的是，我们在枚举最后一个货仓管辖的范围时，计算的距离之和未必是最小的，比如我们在求f[5][3]时，枚举最后一个货仓管辖范围是最后两个房子时，也就是说前两个货仓管辖1-3的房子，第三个货仓管辖4-5的房子，那么最后一个货仓肯定放在4到5之间，但是并不能保证第4个房子到第二个货仓的距离大于到第三个货仓的距离，也就是说，我们计算出来的距离之和不一定是最小的。这相当于局部不是最优的，但是整体是最优的，我们不需要遍历所有的候选解，只需要在包含最优解的集合里找到最优解即可。尽管单次状态转移计算的距离之和可能未必准确，但是最优解却是包含在我们的状态转移之中的。比如说最优解是1号货仓管辖1号房子，2号货仓管辖2 3号房子，3号货仓管辖4 5号房子，我们在进行状态转移时肯定会包含这种状态的划分。

最后说下易错点，初始状态除了f[0][0] = 0外其他状态都要初始化为INF，因为要求的是最小距离和。习惯了memset成0x3f，但是如果直接使用vector的构造函数初始化，每个状态的值要初始化为0x3f3f3f3f。

代码

class Solution {
public:
    int minDistance(vector<int>& houses, int k) {
        int n = houses.size();
        sort(houses.begin(),houses.end());
        vector<vector<int> > f(n + 1,vector<int>(k + 1,0x3f3f3f3f));
        vector<vector<int> > c(n + 1,vector<int>(n + 1,0));
        for(int len = 2;len <= n;len++) {
            for(int i = 1;i <= n;i++) {
                int j = i + len - 1;
                if(j > n)   continue;
                c[i][j] = c[i+1][j-1] + houses[j-1] - houses[i-1];
            }
        }
        f[0][0] = 0;
        for(int i = 1;i <= n;i++) {
            for(int j = 1;j <= k && j <= i;j++) {
                for(int t = j - 1;t < i;t++) {
                    f[i][j] = min(f[i][j],f[t][j-1] + c[t+1][i]);
                }
            }
        }
        return f[n][k];
    }
};