算法刷题 -- 73. 矩阵置零<难度 ★★☆>

1、

    思路:
  1. 标记数组,行和列为0的位置
  2. 然后再统一根据为0的行和列,将对应的行和列数字置零
class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        int[] row = new int[m];
        int[] col = new int[n];
        
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (matrix[i][j] != 0) {
                    continue;
                }
                row[i] = 1;  // 做标记
                col[j] = 1; // 标记
            }
        }

        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (row[i] == 1) { // 根据标记置零
                    matrix[i][j] = 0;
                }
                if (col[j] == 1) { // 根据标记置零
                    matrix[i][j] = 0;
                }
            }
        }
    }
}
经验分享 程序员 微信小程序 职场和发展