面试题 01.08. 零矩阵

面试题 01.08. 零矩阵

编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。

 

示例 1：

输入：
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
输出：
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

示例 2：

输入：
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
输出：
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

---

基本思路：用两个数组标记为零的行和列，空间复杂度O(m+n);

    void setZeroes(vector<vector<int>>& matrix) {
        vector<int> rows(matrix.size(),1);
        vector<int> cols(matrix[0].size(),1);
        for(int i=0;i<matrix.size();i++){

            for(int j=0;j<matrix[0].size();j++){
                if(matrix[i][j]==0){
                    rows[i]=0;
                    cols[j]=0;
                }
            }
        }

        for(int i=0;i<matrix.size();i++){

            for(int j=0;j<matrix[0].size();j++){
                if(rows[i]*cols[j]==0){
                    matrix[i][j]=0;
                }
            }
        }


    }

基本思路：首先用两个标记首行和首列中是否含有零，再用首行和首列标记所在的行或者列中是否含有零  空间复杂度O(1)

    void setZeroes(vector<vector<int>>& matrix) {
        bool isFirstRows=false;
        bool isFirstCols=false;
        for(int i=0;i<matrix.size();i++){
            if(matrix[i][0]==0){
                isFirstCols=true;
                break;
            }
        }

        for(int j=0;j<matrix[0].size();j++){
            if(matrix[0][j]==0){
                isFirstRows=true;
                break;
            }
        }
        for(int i=0;i<matrix.size();i++){
            for(int j=0;j<matrix[0].size();j++){
                if(matrix[i][j]==0){
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                }
            }
        }

        for(int i=1;i<matrix.size();i++){

            for(int j=1;j<matrix[0].size();j++){
                if(matrix[0][j]==0||matrix[i][0]==0){
                    matrix[i][j]=0;
                }
            }
        }
        if(isFirstCols){
            for(int i=0;i<matrix.size();i++)
                matrix[i][0]=0;      
        }
        if(isFirstRows){
            for(int j=0;j<matrix[0].size();j++)
                matrix[0][j]=0;      
        }
    }