【力扣面试】面试题 01.08. 零矩阵
题目
编写一种算法,若M × N矩阵中某个元素为0,则将其所在的行与列清零。
示例 1:
输入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
示例 2:
输入:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
输出:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
解题思路
思路:
1、先找到数组中为0的位置(i,j)然后将ij存储在栈中
2、将栈中内容依次pop两个,(i,j)将i行,和j列内容置0
class Solution {
public void setZeroes(int[][] matrix) {
int row = matrix.length; //代表行的大小 M
int col = matrix[0].length; //代表列的大小 N
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < row; i++) {
for (int j = 0; j < col; j++) {
if (matrix[i][j] == 0) {
stack.push(i);
stack.push(j);
}
}
}
while (!stack.isEmpty()) {
int j = stack.pop(); //取列
int i = stack.pop(); //取行
for (int k = 0; k < row; k++) {
matrix[k][j] = 0;
}
for (int k = 0; k < col; k++) {
matrix[i][k] = 0;
}
}
}
}