本文共 785 字,大约阅读时间需要 2 分钟。
题目描述
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。进阶:
一个直观的解决方案是使用 O(mn) 的额外空间,但这并不是一个好的解决方案。
一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。 你能想出一个仅使用常量空间的解决方案吗? 输入:matrix = [[1,1,1],[1,0,1],[1,1,1]] 输出:[[1,0,1],[0,0,0],[1,0,1]]来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/set-matrix-zeroes 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。c++
class Solution { /* 起初我的思路是: 设置一个二维向量,遍历一遍矩阵,记录所有0元素的坐标,这种思路的空间复杂度是O(m+n) 题目要求常量时间复杂度:借用矩阵的首行首列,但是首行首列中也可能 会有0元素,那就设置两个布尔型变量,首先记录首行首列是否需要置0.最后使用这两个变量来处理首行首列 在实际代码中,我们首先预处理出两个标记变量,接着使用其他行与列去处理第一行与第一列,然后反过来使用第一行与第一列去更新其他行与列,最后使用两个标记变量更新第一行与第一列即可。 */public: void setZeroes(vector>& matrix) { bool first_row=false; bool first_col=false; //先给两个变量赋值,因为之后首行首列会变 for(int i=0;i