Hot-100-Algorithm/矩阵/矩阵置零/main.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
			}
		}
	}
}