func setZeroes(matrix [][]int) { m, n := len(matrix), len(matrix[0]) rows := make([]bool, m) cols := make([]bool, n) for i, row := range matrix { for j, v := range row { if v == 0 { rows[i] = true cols[j] = true } } } for i := 0; i < m; i++ { for j := 0; j < n; j++ { if rows[i] || cols[j] { matrix[i][j] = 0 } } } }