LeetCode 73 矩阵置0
给定一个m x n的矩阵d,如果d[i][j]=0,则矩阵第i行j列全部要置为0,题目要求使用常量的额外空间(不能m*n或者m+n)
思路:使用矩阵第一行和第一列存储对应的行列是否要置0,再用两个额外的变量,标识第一行第一列本身是否要置0
func setZeroes(matrix [][]int) {
iscol0, isrow0 := false, false
for i := 0; i < len(matrix); i++ {
if matrix[i][0] == 0 {
iscol0 = true
break
}
}
for i := 0; i < len(matrix[0]); i++ {
if matrix[0][i] == 0 {
isrow0 = true
break
}
}
for i := 1; i < len(matrix); i++ {
for j := 1; j < len(matrix[0]); j++ {
if matrix[i][j] == 0 {
matrix[0][j], matrix[i][0] = 0, 0
}
}
}
for i := 1; i < len(matrix); i++ {
for j := 1; j < len(matrix[0]); j++ {
if matrix[i][0] == 0 || matrix[0][j] == 0 {
matrix[i][j] = 0
}
}
}
if iscol0 {
for i := 0; i < len(matrix); i++ {
matrix[i][0] = 0
}
}
if isrow0 {
for i := 0; i < len(matrix[0]); i++ {
matrix[0][i] = 0
}
}
}