题目连接:Leetcode 073 Set Matrix Zeroes
解题思路:原理是用一行和一列作为标记,然后用单独的两个变量记录它们是否被归零。我是用matrix[0][x]和matrix[y][0]作标记,遍历一遍数组,如果matrix[i][j]为零,那么标记matrix[0][j] = matrix[i][0] = 0。然后最后再遍历一遍数组,修改值,如果matrix[0][j] 或 matrix[i][0] = 0,则令matrix[i][j] = 0。
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
if (matrix.size() == 0 || matrix[0].size() == 0) return ;
int m = matrix.size(), n = matrix[0].size();
bool row = false, col = false;
for (int i = 0; i < n; i++) if (matrix[0][i] == 0) row = true;
for (int i = 0; i < m; i++) if (matrix[i][0] == 0) col = true;
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if (matrix[i][j] == 0)
matrix[i][0] = matrix[0][j] = 0;
}
}
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if (matrix[i][0] == 0 || matrix[0][j] == 0)
matrix[i][j] = 0;
}
}
if (row) for (int i = 0; i < n; i++) matrix[0][i] = 0;
if (col) for (int i = 0; i < m; i++) matrix[i][0] = 0;
}
};