首页 > 代码库 > Set Matrix Zeroes 将矩阵置零
Set Matrix Zeroes 将矩阵置零
给定一个矩阵，把零值所在的行和列都置为零。例如：
1 2 3
1 0 3
1 1 1
操作之后变为
1 0 3
0 0 0
1 0 1
方法1：
赋值另存一个m*n的矩阵，在原矩阵为零的值相应置新的矩阵行和列为零。额外空间为O(m*n).
方法2：
两个数组，bool[m] 和 bool[n] 分别存某行有零，后者某列有零。之后根据数组值将原矩阵相应位置置零。额外空间O(m + n)。
class Solution {public: void setZeroes(vector<vector<int> > &matrix) { int len1 = matrix.size(); if (len1 == 0) return ; int len2 = matrix[0].size(); if (len2 == 0) return ; vector<bool > row(len1), col(len2); for (int i = 0; i < len1; i++) for (int j = 0; j < len2; j++) { if (matrix[i][j] == 0) { row[i] = true; col[j] = true; } } for (int i = 0; i < len1; i++) for (int j = 0; j < len2; j++) { if (row[i] == true) matrix[i][j] = 0; else if (col[j] == true) matrix[i][j] = 0; } return ; }};
方法3：（常数额外空间）
1. 找到一个零的位置，把这行这列当做方法2中的两个数组存值。
2. 根据1的位置的所在行和列的值是否有零将矩阵相应位置置零。
3. 再把1中零所在位置的行和列置零。
class Solution {public: void setZeroes(vector<vector<int> > &matrix) { int len1 = matrix.size(); if (len1 == 0) return ; int len2 = matrix[0].size(); if (len2 == 0) return ; int row = -1, col = -1; for (int i = 0; i < len1; i++) for (int j = 0; j < len2; j++) { if (matrix[i][j] == 0) { row = i; col = j; } } if (row == -1) return; for (int i = 0; i < len1; i++) for (int j = 0; j < len2; j++) { if (matrix[i][j] == 0 && i != row && j != col) { matrix[i][col] = 0; matrix[row][j] = 0; } } for (int i = 0; i < len1; i++) for (int j = 0; j < len2; j++) { if (i != row && j != col) { if (matrix[i][col] == 0 ) matrix[i][j] = 0; else if (matrix[row][j] == 0) matrix[i][j] = 0; } } int index = -1; while(++index < len1) matrix[index][col] = 0; index = -1; while(++index < len2) matrix[row][index] = 0; return ; }};
Set Matrix Zeroes 将矩阵置零
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。