LeetCode 1253. 重構 2 行二進制矩陣 - Java - 統計

題目鏈接：https://leetcode-cn.com/contest/weekly-contest-162/problems/reconstruct-a-2-row-binary-matrix/

---

給你一個 2 行 n 列的二進制數組：

矩陣是一個二進制矩陣，這意味着矩陣中的每個元素不是 0 就是 1。
第 0 行的元素之和爲 upper。
第 1 行的元素之和爲 lower。
第 i 列（從 0 開始編號）的元素之和爲 colsum[i]，colsum 是一個長度爲 n 的整數數組。
你需要利用 upper，lower 和 colsum 來重構這個矩陣，並以二維整數數組的形式返回它。

如果有多個不同的答案，那麼任意一個都可以通過本題。

如果不存在符合要求的答案，就請返回一個空的二維數組。

示例 1：

輸入：upper = 2, lower = 1, colsum = [1,1,1]
輸出：[[1,1,0],[0,0,1]]
解釋：[[1,0,1],[0,1,0]] 和 [[0,1,1],[1,0,0]] 也是正確答案。

示例 2：

輸入：upper = 2, lower = 3, colsum = [2,2,1,1]
輸出：[]

示例 3：

輸入：upper = 5, lower = 5, colsum = [2,1,2,0,1,0,1,2,0,1]
輸出：[[1,1,1,0,1,0,0,1,0,0],[1,0,1,0,0,0,1,1,0,1]]

提示：

• 1 <= colsum.length <= 10^5
• 0 <= upper, lower <= colsum.length
• 0 <= colsum[i] <= 2

---

題解

看了題目之後，我發現可以用回溯來做，然而並非如此。通過提示，我們可以看到數據是非常多的。達到了 $10^5$，這說明這種方法是會超時的。仔細讀題目後，看到了新的關鍵字。如果有多個不同的答案，任意一個即可。

數組 colsum 是很特別的，它的和就是我們要得到的矩陣的所有數據的和。而 upper 和 lower 的和也是矩陣的和。

因此，可以先來統計一下 1 和 2 的個數。

如果數據是合法的。再依次將 0 或 1 添加到鏈表中即可。

---

Java代碼

/**
 * Description: Reconstruct a 2-Row Binary Matrix
 * Author: wowpH
 * Date: 2019-11-11 17:03:33
 * From: https://www.cnblogs.com/wowpH/p/11836688.html
 */
class Solution {
    public List<List<Integer>> reconstructMatrix(int upper, 
                                                 int lower, 
                                                 int[] colsum) {
        int n = colsum.length;
        List<List<Integer>> ret = new ArrayList<>();
        // 統計1，2的個數
        int one = 0, two = 0;
        for (int i = 0; i < n; ++i) {
            if (1 == colsum[i]) {
                ++one;
            } else if (2 == colsum[i]) {
                ++two;
            }
        }
        // 檢查參數是否合法
        if (one + (two << 1) != upper + lower) {
            return ret;
        }
        // 提前去掉都是1的個數
        upper -= two;
        lower -= two;
        if (upper < 0 || lower < 0) {
            return ret;
        }
        // 添加0，1到鏈表中
        List<Integer> list1 = new ArrayList<>();
        List<Integer> list2 = new ArrayList<>();
        for (int i = 0; i < n; ++i) {
            if (0 == colsum[i]) {
                // 都添加0
                list1.add(0);
                list2.add(0);
            } else if (2 == colsum[i]) {
                // 都添加1
                list1.add(1);
                list2.add(1);
            } else {
                // 1個0，1個1
                if (upper > 0) {
                    // 鏈表1還可以添加1，優先滿足鏈表1
                    list1.add(1);
                    list2.add(0);
                    --upper;
                } else {
                    // 鏈表1已經不能添加1，向鏈表2添加1
                    list1.add(0);
                    list2.add(1);
                    --lower;
                }
            }
        }
        // 添加到結果鏈表中並返回結果
        ret.add(list1);
        ret.add(list2);
        return ret;
    }
}

---

原文鏈接：https://www.cnblogs.com/wowpH/p/11836688.html

---

- End -