题目
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
进阶:
一个直观的解决方案是使用 O(mn) 的额外空间,但这并不是一个好的解决方案。
一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。
你能想出一个仅使用常量空间的解决方案吗?
示例
输入:matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出:[[1,0,1],[0,0,0],[1,0,1]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/set-matrix-zeroes
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
想法
1、遍历二维数组,存储每个0的位置。resi 存 i,resj 存 j。
2、把 resi 中每一行变为0。把resj中每一列变为0。
实现
方法1:python
class Solution:
def setZeroes(self, matrix: List[List[int]]) -> None:
"""
Do not return anything, modify matrix in-place instead.
"""
resi,resj=list(),list()
row=len(matrix)
col=len(matrix[0])
for i in range(row):
for j in range(col):
if matrix[i][j]==0:
resi.append(i)
resj.append(j)
for i in resi:
for j in range(col):
matrix[i][j]=0
for j in resj:
for i in range(row):
matrix[i][j]=0
方法2:java
class Solution {
public void setZeroes(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
// 记录0元素的行列
Set<Integer> row = new HashSet<>();
Set<Integer> cln = new HashSet<>();
// 寻找0元素
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (matrix[i][j] == 0) {
row.add(i);
cln.add(j);
}
}
}
// 行置0
for (Integer i : row) {
for (int j = 0; j < n; j++) {
matrix[i][j] = 0;
}
}
// 列置0
for (Integer j : cln) {
for (int i = 0; i < m; i++) {
matrix[i][j] = 0;
}
}
}
}