class Solution { public void setZeroes(int[][] matrix) { int m = matrix.length, n = matrix[0].length; boolean i0 = false, j0 = false; for (int j = 0; j < n; ++j) { if (matrix[0][j] == 0) { i0 = true; break; } } for (int i = 0; i < m; ++i) { if (matrix[i][0] == 0) { j0 = true; break; } } for (int i = 1; i < m; ++i) { for (int j = 1; j < n; ++j) { if (matrix[i][j] == 0) { matrix[i][0] = 0; matrix[0][j] = 0; } } } for (int i = 1; i < m; ++i) { for (int j = 1; j < n; ++j) { if (matrix[i][0] == 0 || matrix[0][j] == 0) { matrix[i][j] = 0; } } } if (i0) { for (int j = 0; j < n; ++j) { matrix[0][j] = 0; } } if (j0) { for (int i = 0; i < m; ++i) { matrix[i][0] = 0; } } } }