给你一个 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
思路:
也不需要什么算法,
直接模拟即可,
首先生成一个全为0的结果数组 b,
优先找到所有能使colsum[i] 为 2 的 i,然后把b[0][i] = 1, b[1][i] = 1, 然后更新upper和lower,
接着再次遍历colsum,处理colsum[i] 为1的部分,因为只需输出任意可能解,所以顺序不重要。
最后判断一下是不是恰好能用输入里的upper 和 lower 构建满足colsum 的结果数组b。
时间复杂度:O(len(colsum))
空间复杂度:O(1)
class Solution(object):
def reconstructMatrix(self, upper, lower, colsum):
"""
:type upper: int
:type lower: int
:type colsum: List[int]
:rtype: List[List[int]]
"""
b = [[0 for _ in range(len(colsum))] for _ in range(2)]
for i, cs in enumerate(colsum):
if cs == 2:
b[0][i] = 1
b[1][i] = 1
upper -= 1
lower -= 1
for i, cs in enumerate(colsum):
if cs == 1:
if lower > upper:
b[1][i] = 1
lower -= 1
else:
b[0][i] = 1
upper -= 1
if upper != 0 or lower != 0: #没有解
return []
return b