郑州网站建设学校南宁网站推广排名

题目

给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。

示例

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,1]]

解析

这道题题目上要求用原地算法，先说开辟额外辅助空间的方法吧，原理上就是对于行和列开辟两个布尔型数组，先遍历一遍矩阵，将为0的位置（行和列的下标）置为0，完事后再遍历一遍矩阵，把对应位置的值置为0即可。

func setZeroes(matrix [][]int) {row := make([]bool, len(matrix))col := make([]bool, len(matrix[0]))for i, r := range matrix {for j, v := range r {if v == 0 {row[i] = truecol[j] = true}}}for i, r := range matrix {for j := range r {if row[i] || col[j] {r[j] = 0}}}
}

另外还有不适用额外空间，只需要定义变量的写法，但感觉有点复杂了，先不写了