LeetCode 面试题 01.08 零矩阵[记忆化] HERODING的LeetCode之路

Posted 2022-09-30 HERODING23

解题思路：
首先定义两个行列数组，用来判断某一行或者某一列是否有零，遍历一遍矩阵后更新两个数组，再分别遍历两个数组，对直接进行整行和整列的操作，代码如下：

class Solution 
public:
    void setZeroes(vector<vector<int>>& matrix) 
        int m = matrix.size();
        int n = matrix[0].size();
        vector<bool> row(m, true);
        vector<bool> col(n, true);
        for(int i = 0; i < m; i ++) 
            for(int j = 0; j < n; j ++) 
                if(matrix[i][j] == 0) 
                    row[i] = false;
                    col[j] = false;
                
            
        
        for(int i = 0; i < m; i ++) 
            if(!row[i]) 
                matrix[i].assign(matrix[i].size(), 0);
            
        
        for(int i = 0; i < n; i ++) 
            if(!col[i]) 
                for(int j = 0; j < m; j ++) 
                    matrix[j][i] = 0;
                
            
        
    
;