1595. 连通两组点的最小成本

2021/09/21 leetcode 共 1249 字,约 4 分钟

title

problem link

给你两组点,其中第一组中有 size1 个点,第二组中有 size2 个点,且 size1 >= size2 。

任意两点间的连接成本 cost 由大小为 size1 x size2 矩阵给出,其中 cost[i][j] 是第一组中的点 i 和第二组中的点 j 的连接成本。如果两个组中的每个点都与另一组中的一个或多个点连接,则称这两组点是连通的。换言之,第一组中的每个点必须至少与第二组中的一个点连接,且第二组中的每个点必须至少与第一组中的一个点连接。

返回连通两组点所需的最小成本。

示例 1:

img

输入:cost = [[15, 96], [36, 2]]
输出:17
解释:连通两组点的最佳方法是:
1--A
2--B
总成本为 17 。

示例 2:

img

输入:cost = [[1, 3, 5], [4, 1, 1], [1, 5, 3]]
输出:4
解释:连通两组点的最佳方法是:
1--A
2--B
2--C
3--A
最小成本为 4 。
请注意,虽然有多个点连接到第一组中的点 2 和第二组中的点 A ,但由于题目并不限制连接点的数目,所以只需要关心最低总成本。

示例 3:

输入:cost = [[2, 5, 1], [3, 4, 7], [8, 1, 2], [6, 2, 4], [3, 8, 8]]
输出:10

提示:

  • size1 == cost.length
  • size2 == cost[i].length
  • 1 <= size1, size2 <= 12
  • size1 >= size2
  • 0 <= cost[i][j] <= 100

solution

class Solution:
    def connectTwoGroups(self, cost: List[List[int]]) -> int:
        n1 = len(cost)
        n2 = len(cost[0])
        inf = float('inf')
        mcost = [[0] * (1 << n2) for _ in range(n1)]
        for i in range(n1):
            for j in range(1, 1 << n2):
                mcost[i][j] = sum(cost[i][k] for k in range(n2) if (1 << k) & j)
        import copy
        dp = copy.deepcopy(mcost[0])
        for i in range(1, n1):
            new = [inf] * (1 << n2)
            for mask in range((1 << n2) - 1, 0, -1):
                # 确保当前被选了
                for j in range(n2):
                    bit = 1 << j
                    new[mask | bit] = min(new[mask | bit], dp[mask] + mcost[i][bit])
                # 如果不选的话,肯定是从剩下的里面选个子集加上去
                rest = ((1 << n2) - 1) ^ mask
                sub = rest
                while sub:
                    new[mask | sub] = min(new[mask | sub], dp[mask] + mcost[i][sub])
                    sub = (sub - 1) & rest
            dp = new
        return dp[(1 << n2) - 1]

Search

    Table of Contents