28 lines
493 B
Go
28 lines
493 B
Go
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
|
|
}
|
|
}
|
|
}
|
|
}
|