矩阵置零
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
示例 1:
输入:matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出:[[1,0,1],[0,0,0],[1,0,1]]
示例 2:
输入:matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出:[[0,0,0,0],[0,4,5,0],[0,3,1,0]]
提示:
m == matrix.length
n == matrix[0].length
1 <= m, n <= 200
-231 <= matrix[i][j] <= 231 - 1
进阶:
一个直观的解决方案是使用 O(mn) 的额外空间,但这并不是一个好的解决方案。
一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。
你能想出一个仅使用常量空间的解决方案吗?
暴力求解:
class Solution { public: void setZeroes(vector<vector<int>>& matrix) { vector<pair<int,int>> zero_index; for(int i=0;i<matrix.size();i++){ for(int j=0;j<matrix[i].size();j++){ if(matrix[i][j] == 0 ){ zero_index.push_back(make_pair(i,j)); } } } for(auto x:zero_index){ for(int i=0;i<matrix[0].size();i++){ matrix[x.first][i] = 0; } for(int i=0;i<matrix.size();i++){ matrix[i][x.second] = 0; } } } }; bool firstColHasZero = false; bool firstRowHasZero = false; for(int i=0;i<matrix[0].size();i++){ if(matrix[0][i] == 0){ firstRowHasZero = true; break; } }for(int i=0;i<matrix.size();i++){ if(matrix[i][0] == 0){ firstColHasZero = true; break; } }
for(int i=1;i<matrix.size();i++){ for(int j=1;j<matrix[0].size();j++){ if(matrix[i][j] == 0){ matrix[0][j] = 0; matrix[i][0] = 0; } } }
for(int i=1;i<matrix[0].size();i++){ if(matrix[0][i] == 0){ for(int j=1;j<matrix.size();j++){ matrix[j][i] = 0; } } }
for(int i=1;i<matrix.size();i++){ if(matrix[i][0] == 0){ for(int j=1;j<matrix[0].size();j++){ matrix[i][j] = 0; } } }
if(firstColHasZero){ for (size_t i = 0; i < matrix.size(); i++) { matrix[i][0] = 0; } }
if(firstRowHasZero){ for (size_t i = 0; i < matrix[0].size(); i++) { matrix[0][i] = 0; } } 关键: 1.暴力算法记录0的index,去让这一行和这一列为0 2.原地算法,第一行和第一列记录有0的位置,然后处理所有行列,最后处理第一行列为0的情况 标签:10.25,示例,int,解决方案,matrix,算法,size From: https://www.cnblogs.com/minipython-wldx/p/17786680.html