2023.2.17每日一题——面试题 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]
]

---

解题核心

找到要处理的行和列，并且要想办法标记，避免同一行/列多次清零的操作

解法一：模拟

找到要清零的行列然后进行操作

public void setZeroes(int[][] matrix) {
    int row = matrix.length;//行
    int col = matrix[0].length;//列
    HashSet<Integer> rowSet = new HashSet<>();
    HashSet<Integer> colSet = new HashSet<>();
    //记录要清零的行和列(去重减少操作)
    for (int i = 0; i < row * col; i++) {
        if (matrix[i % row][i / row] == 0) {
            rowSet.add(i % row);
            colSet.add(i / row);
        }
    }
    //清零
    for (int index : rowSet) {
        for (int i = 0; i < col; i++) {
            matrix[index][i] = 0;
        }
    }
    for (int index : colSet) {
        for (int i = 0; i < row; i++) {
            matrix[i][index] = 0;
        }
    }
}