Leetcode之Set Matrix Zeroes 问题

来源:互联网 发布:淘宝客佣金终于追回了 编辑:程序博客网 时间:2024/06/05 20:46

问题描述:

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place. 

Follow up:

Did you use extra space?
A straight forward solution using O(mn) space is probably a bad idea.
A simple improvement uses O(m + n) space, but still not the best solution.
Could you devise a constant space solution?

问题来源:Set Matrix Zeroes (详细地址:https://leetcode.com/problems/set-matrix-zeroes/description/)

思路分析:题目的意思就是让我们找到数组中的元素,并将它们所在的行和列都设为0,而且题目说的只能是原地解决,不能额外申请空间,下面解释下空间复杂度为O(mn)的解法和O(m + n)的解法该咋解:

O(mn):非常明显,就是和原来的数组一一对应,设置为0也是一个一个地设置,题目说了这是一种bad的想法;

O(m + n):另外我们再申请一行和一列,用来存放这行和这列是否存在0元素,如果存在的话,那么我们就将对应的行和列上的这一格设置为0,有点标记的意思;

O(1):回到我们这道题,我们可以从O(m + n)的想法中得到启示,我们可以将第0行和第0列设置为标记行和列,这样我们就可以处理其他元素了,可能有人问了?那第一行第一列咋办呢?他们本身有0和没0咋区分呢?我们采用两个布尔变量,firstRow和firstCol来标记第一行.第一列的情况不就完了吗?处理好别的行别的列之后我们回过头再来处理第0行和第0列。接下来我们就试试吧。

代码:

第一行和第一列拿出来标记:

其他行标记和设零的过程:


第一行和第一列设置为0: