leetcode1478. 安排邮筒（Python3）

leetcode1478. 安排邮筒

给你一个房屋数组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^4
• 1 <= k <= n
• 数组 houses 中的整数互不相同。

---

方法：动态规划

思路：

首先需要对houses进行排序，这样房子的顺序按照位置来排序。

我们先考虑一个问题，如果只有一个邮筒，那么放在这些哪个位置会使得距离和最小呢？答案是放在这些房子的中位数的那个房子的位置，这是由绝对值不等式决定的。我们的目标是将所有的房子，分成若干个块，每块放一个邮筒在这个块的中位数位置。求出分成若干块的距离和，最小距离和即为答案。最小距离和时，不一定分成k块使用k个邮筒，可能使用少于k个邮筒也可达到，这时剩下的邮筒就随意摆放即可，不会影响答案，相当于这些邮筒不是离房子最近的。

对于每一块的距离和，我们维护一个cost数组，cost[i] [j]表示房子下标从i到j的这一块，把邮筒放在中位数处的距离和。

我们设定dp数组，dp(i,j)表示下标为0-i的这i+1个房子，最多分为j块时的最小距离和。那么我们的答案应该是dp(n-1,k)。

现在来考虑初始条件，首先dp(i,0)表示分成0块，显然不成立，把dp(i,0)，i取值为[0,n-1]，都置为无穷大。

对于dp(0,i)，i取值为[1,k]，表示对第一个房子，分为若干块，显然只要把一个邮筒放在房子处即可，因此都置为0。

下面考虑状态转移方程，对于dp(i,j)我们考虑分成的最后一块的起始坐标m，m可能的取值为[0,i]。如果m=0，那么这种情况下就是相当于只分一块，距离和为cost[0] [i]；m不等于0的时候，距离和就是dp(m-1,j-1)+cost[m] [i]。即前面m-1房子最多分为j-1块的最小距离和加上最后从m到i的一块的距离和。那么dp(i,j)的值即为所有m取值情况的距离和中的最小值。
d p