LeetCode 每日一题1253. 重构 2 行二进制矩阵

1253. 重构 2 行二进制矩阵

题意

给你一个 2 行 n 列的二进制数组：

矩阵是一个二进制矩阵，这意味着矩阵中的每个元素不是 0 就是 1。
第 0 行的元素之和为 upper。
第 1 行的元素之和为 lower。
第 i 列（从 0 开始编号）的元素之和为 colsum[i]，colsum 是一个长度为 n 的整数数组。
你需要利用 upper，lower 和 colsum 来重构这个矩阵，并以二维整数数组的形式返回它。

如果有多个不同的答案，那么任意一个都可以通过本题。

如果不存在符合要求的答案，就请返回一个空的二维数组。

样例输入

upper = 2, lower = 1, colsum = [1,1,1]

样例输出

[[1,1,0],[0,0,1]]

提示

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

c++代码

class Solution {
public:
    vector<vector<int>> reconstructMatrix(int upper, int lower, vector<int>& colsum) {
        if(upper + lower != accumulate(colsum.begin(), colsum.end(), 0)) return {};
        vector<vector<int>> ans(2, vector<int>(colsum.size()));
        for(int i = 0; i < colsum.size(); i++ ){
            if(colsum[i] == 2){
                if(upper <= 0 || lower <= 0) return {};
                ans[0][i] = ans[1][i] = 1;
                upper--;
                lower--;
            }
            if(colsum[i] == 1){
                if(upper <= 0 && lower <= 0) return {};
                if(upper > lower){
                    ans[0][i] = 1;
                    upper--;
                }else{
                    ans[1][i] = 1;
                    lower--;
                }
            }
        }
        return ans;
    }
};

python代码

class Solution:
    def reconstructMatrix(self, upper: int, lower: int, colsum: List[int]) -> List[List[int]]:
        n = len(colsum)
        totalSum = 0
        for item in colsum:
            totalSum += item;
        if totalSum != upper + lower: return []
        ans = [[0] * n for _ in range(2)]

        for i in range(0, n):
            if colsum[i] == 2:
                if upper < 1 or lower < 1: return []
                ans[0][i] = ans[1][i] = 1
                upper -= 1
                lower -= 1
            if colsum[i] == 1:
                if upper < 1 and lower < 1: return []
                if upper > lower:
                    ans[0][i] = 1
                    upper -= 1
                else:
                    ans[1][i] = 1
                    lower -= 1
        return ans

备注

c++ 中accumulate函数可以求一个数组的和，accumulate(a.begin(), b.begin(), 0)
c++ 中使用vector容器创建二维数组为vector<vector<int>>a(2, vector<int>(n))
python 中创建二维数组的方法 a = [[0]*n for _ in range(n)]