【题目描述】Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.
click to show follow up.
Follow up:
Did you use extra space?
A straight forward solution using O(m n) space is probably a bad idea.
A simple improvement uses O(m + n) space, but still not the best solution.
Could you devise a constant space solution?
【解题思路】使用两个数组来标记,这个数组就是标记,某一行或是某一列是否存在 0。与矩阵对应的是两个标记数组,数组中,如果此行或是此列中存在 0 就置为 0,如果不存在 0 就置为 1。
【考查内容】数组,查找
class Solution {
public:
void setZeroes(vector<vector<int> > &matrix) {
int rawLength = matrix.size();
int colLength = matrix[0].size();
vector<bool> rawFlags(rawLength);
vector<bool> colFlags(colLength);
for (int i = 0; i < rawLength; i++) {
for (int j = 0; j < colLength; j++) {
if (matrix[i][j] == 0) {
rawFlags[i] = true;
colFlags[j] = true;
}
}
}
// 调整每一行置0
for (int i = 0; i < rawLength; i++) {
if (rawFlags[i]) {
for (int j = 0; j < colLength; j++) {
matrix[i][j] = 0;
}
}
}
// 调整每一列置0
for (int i = 0; i < colLength; i++) {
if (colFlags[i]) {
for (int j = 0; j < rawLength; j++) {
matrix[j][i] = 0;
}
}
}
}
};