leetcode-面试题 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]
]

这道题我们可能会首先想到遍历数组,遍历到0的位置就将对应的矩阵的行和列的元素归零.但是我们却忽略了一个问题,归0后的元素我们后面还会重复遍历到,所以会导致最终整个矩阵都变0

为了解决这个问题,我们可以建立一个和matrix同等大小的bool数组进行标记,最后统计完再利用这个数组修改原数组

我们同样可以考虑到,当某一元素为0的时候,该元素所在的行列都要归零,所以它的同行同列元素后面就不需要再进行考虑了,所以我们可以换个标记数组,利用两个bool数组row和column分别标记0元素对应的行和列为true,最后再进行处理.

我的代码：

C++

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {

        vector<bool> row(matrix.size());
        vector<bool> column(matrix[0].size());
     
        for (int i = 0; i < matrix.size(); ++i){
            for (int j = 0; j < matrix[0].size(); ++j){
                if (matrix[i][j] == 0){
                    row[i] = true;
                    column[j] = true;

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

    }
};

Java

public class Solution {
    public void SetZeroes(int[][] matrix) {
    bool[] row = new bool[matrix.Length];
    bool[] column = new bool[matrix[0].Length];
    for(int i=0;i<matrix.Length;i++){
        for(int j=0;j<matrix[0].Length;j++){
            if(matrix[i][j]==0){
                row[i]=true;
                column[j]=true;
            }
        }
    }
    for(int i=0;i<matrix.Length;i++){
        for(int j=0;j<matrix[0].Length;j++){
            if(row[i]||column[j]){
                matrix[i][j]=0;
            }
        }
    }

    }
}