数据结构每日一题10

给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。

使用两个数组记录每行，每列有无为零的元素，在遍历一次将有零的行列全置为零，需要遍历两次矩阵

class Solution {

public:

    void setZeroes(vector<vector<int>>& matrix) {

        int m = matrix.size();

        int n = matrix[0].size();

        vector<int> hang(m); vector<int> lie(n);

        for(int i =0; i<m;i++){

            for(int j = 0;j<n;j++){

                if(matrix[i][j] == 0){

                    hang[i] = 1;lie[j]=1;

                }

            }

        }

        for(int i = 0;i<m;i++){

            for(int j=0;j<n;j++){

                if(hang[i] || lie[j]){

                    matrix[i][j] = 0;

                }

            }

        }

    }

};