leetcode73. 矩阵置零

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

AC代码

class Solution {
public:
    void setZeroes(vector>& matrix) {
        unordered_set row;
        unordered_set 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; 
                }
            }
        }
    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

代码解释
给定一个二维整数矩阵，当矩阵中的元素值为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。