一个矩阵,如果位置
开一个row[]
和col[]
分别记录哪些行,哪些列需要置为0即可。
从上一个算法我们可以知道必须要保存需要update的行和列,但是只能额外开
于是,类似于十字交叉链表的思想,我们可以用第一行和第一列来记录需要update的行和列。
如下图所示:
在算法2中,我们需要注意这样一个问题,我们用第一行的某一个元素r
和c
来记录第一行和第一列的更新情况。
algorithm 1
class Solution {public: void setZeroes(vector<vector<int>>& matrix) { if (matrix.size()) { int m = matrix.size(), n = matrix[0].size(); vector<int> row(matrix.size(), 0); vector<int> col(matrix[0].size(), 0); for (int i = 0; i < m; i++) { for (int j = 0; j < n; j++) { if (!matrix[i][j]) {row[i] = 1, col[j] = 1;} } } for (int i = 0; i < m; i++) { if (row[i]) { for (int j = 0; j < n; j++) matrix[i][j] = 0; } } for (int j = 0; j < n; j++) { if (col[j]) { for (int i = 0; i < m; i++) matrix[i][j] = 0; } } } }};algorithm 2
class Solution {public: void setZeroes(vector<vector<int>>& matrix) { if (matrix.size()) { int m = matrix.size(), n = matrix[0].size(); int r = 0, c = 0; for (int i = 0; i < m; i++) { for (int j = 0; j < n; j++) { if (!matrix[i][j]) { j ? matrix[0][j] = 0 : c = 1; i ? matrix[i][0] = 0 : r = 1; } } } for (int j = 1; j < n; j++) { if (!matrix[0][j]) { for (int i = 1; i < m; i++) matrix[i][j] = 0; } } for (int i = 1; i < m; i++) { if (!matrix[i][0]) { for (int j = 1; j < n; j++) { matrix[i][j] = 0; } } } if (r) { for (int j = 0; j < n; j++) matrix[0][j] = 0; } if (c) { for (int i = 0; i < m; i++) matrix[i][0] = 0; } } }};新闻热点
疑难解答