package main func main() { setZeroes([][]int{{0, 1, 2, 0}, {3, 4, 5, 2}, {1, 3, 1, 5}}) } func setZeroes(matrix [][]int) { col := make([]bool, len(matrix[0])) row := make([]bool, len(matrix)) for i := range matrix { for j := range matrix[i] { if matrix[i][j] == 0 { col[j] = true row[i] = true } } } //fmt.Println("col", col) //fmt.Println("row", row) for i := range matrix { for j := range matrix[i] { if row[i] || col[j] { matrix[i][j] = 0 } } } }