leetcode73. 矩阵置零

nudt_qxx2024-03-05 22:20

链接见：https://leetcode.cn/problems/set-matrix-zeroes/description/
题目描述
给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用原地算法。

AC代码

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        unordered_set<int> row;
        unordered_set<int> 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。