网站建设与设计试题,先做网站先备案,网站开发设计公司块位,公众号投票链接见#xff1a;https://leetcode.cn/problems/set-matrix-zeroes/description/ 题目描述 给定一个 m x n 的矩阵#xff0c;如果一个元素为 0 #xff0c;则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
AC代码
class Solution {
public:void setZeroes(vec…链接见https://leetcode.cn/problems/set-matrix-zeroes/description/ 题目描述 给定一个 m x n 的矩阵如果一个元素为 0 则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
AC代码
class Solution {
public:void setZeroes(vectorvectorint matrix) {unordered_setint row;unordered_setint col;int n matrix.size();int m matrix[0].size();for(int i 0; i n; i) {for (int j 0 ; j m; j) {if(matrix[i][j] 0 ) {row.insert(i);col.insert(j);}}}for(int i 0; i n; i) {for (int j 0 ; j m; j) {if(row.count(i) || col.count(j)) {matrix[i][j] 0; }}}}
};代码解释 给定一个二维整数矩阵当矩阵中的元素值为0时将该元素所在的行和列全部置为0。
逐行解释
定义一个函数setZeroes它接收一个引用类型的二维整数向量matrix作为参数。定义两个unordered_set容器分别用于存储需要置零的行索引row和列索引col。 获取矩阵的行数n和列数m。使用两层循环遍历矩阵中的每个元素。在内层循环中检查当前元素matrix[i][j]是否为0如果是则将行索引i插入到row集合中列索引j插入到col集合中。第二轮两层循环再次遍历矩阵的所有元素。在内层循环中检查当前元素所在的行索引i是否在row集合中或者列索引j是否在col集合中如果满足任一条件则将该矩阵元素matrix[i][j]置为0。