Set Matrix Zeroes

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

要求空间复杂度为o(1)

思路:把数组划分为两部分,第一部分是第一行和第一列;第二部分是除去第一行第一列的矩阵部分。

1.先记录第一行第一列是否需要清零。

2.用第一行来存储第二部分矩阵需要清零的列,第一列来存储第二部分矩阵需要清零的行,例如,如果matrix[i][j] = 0,那么令matrix[i][0]=0,matrix[0][j] = 0,其中i>0,j>0,表示i行需要清零,第j列需要清零。

3.标记需要清0的行列

4.对行列清0

5.根据1,对第一行第一列清零

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size();
        if(m==0){
            return ;
        }
        int n = matrix[0].size();
        
        bool isFirstRowHasZero = false;
        bool isFirstColHasZero = false;
        
        for(int j=0;j<n;j++){
            if(matrix[0][j]==0){
                isFirstRowHasZero = true;
                break;
            }
        }
        
        for(int i=0;i<m;i++){
            if(matrix[i][0]==0){
                isFirstColHasZero = true;
                break;
            }
        }
        
        for(int i=1;i<m;i++){
            for(int j=1;j<n;j++){
                if(matrix[i][j]==0){
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        
        for(int i=1;i<m;i++){
            if(matrix[i][0] == 0){
                for(int j=1;j<n;j++){
                    matrix[i][j] = 0;
                }
            }
        }
        
        for(int j=1;j<n;j++){
            if(matrix[0][j] == 0){
                for(int i=1;i<m;i++){
                    matrix[i][j] = 0;
                }
            }
        }
        
        if(isFirstRowHasZero){
            for(int j=0;j<n;j++){
                matrix[0][j] = 0;
            }
        }
        
        if(isFirstColHasZero){
            for(int i=0;i<m;i++){
                matrix[i][0] = 0;
            }
        }
    }
};
原文地址:https://www.cnblogs.com/zengzy/p/5001829.html