https://leetcode.cn/problems/zero-matrix-lcci/

标记数组

新建布尔数组标记下零元素的位置。

  1. public static void setZeroes(int[][] matrix) {
  2.     boolean[] col = new boolean[matrix.length]; //行
  3.     boolean[] row = new boolean[matrix[0].length]; //列
  5.     for (int i = 0; i < matrix.length; i++) {
  6.         for (int j = 0; j < matrix[0].length; j++) {
  7.             if (matrix[i][j] == 0) {
  8.                 col[i] = true;
  9.                 row[j] = true;
  10.             }
  11.         }
  12.     }
  14.     for (int i = 0; i < matrix.length; i++) {
  15.         for (int j = 0; j < matrix[0].length; j++) {
  16.             if (col[i] || row[j]) {
  17.                 matrix[i][j] = 0;
  18.             }
  19.         }
  20.     }
  22. }