JAVA程序设计：连通两组点的最小成本（LeetCode：1595）

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

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

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

 

示例 1：

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

输入：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

思路：定义dp[i][j]：表示第一个集合中前i个字符和第二个集合状态为j时的最小连接代价。状态表示为二进制，其中第i位为1代表第二个集合中第j个点和第一个集合中某个点相连了。

class Solution {
    public int connectTwoGroups(List<List<Integer>> cost) {

        int ans = 0;
        int n = cost.size();
        int m = cost.get(0).size();
        int[][] dp = new int[n + 1][1 << m];

        for (int i = 0; i <= n; i++)
            Arrays.fill(dp[i], 1000000);

        dp[0][0] = 0;

        for (int i = 1; i <= n; i++)
            for (int j = 0; j < m; j++)
                for (int k = 0; k < (1 << m); k++) {
                    dp[i][k | (1 << j)] = Math.min(dp[i][k | (1 << j)], dp[i][k] + cost.get(i - 1).get(j));
                    dp[i][k | (1 << j)] = Math.min(dp[i][k | (1 << j)], dp[i - 1][k] + cost.get(i - 1).get(j));
                }

        return dp[n][(1 << m) - 1];

    }
}