问题
将一个M*N矩阵中0元素所在行和列清零。
分析
- 注意,不能一次遍历进行处理,如果一次遍历处理最终全部的元素都会变成0,需要首先将0元素的位置标记出来,然后第二遍遍历的时候再处理;
- 一种可行的方式是,可以用一个大小相同的布尔数组来记录0元素出现的位置,但是这样的空间复杂度是o(nm),其实我们不需要记录这么多值,我们仅关注0元素所在的位置,甚至我们仅关注0元素所在的行和列,因为最终不管这些行或者列上有多少个0元素都会被置为0。
- 所以最优的方案是,定义两个数组用来记录哪些行或者哪些列包含了0元素,第二遍遍历的时候根据行或者列包含0元素的情况进行置0。下面的实现中采用布尔类型数组进行标记,如果想再缩小空间可以利用位来表示,注意主要是标识两种状态的情况,我们都可以使用位来标记以达到缩小空间的目的。但是在实际的工程中,为了代码的易读性,且不会对空间有这么苛刻的要求情况下我们用布尔类型足够。
实现
/**
* 1.7问题:将一个M*N矩阵中0元素所在行和列清零
* 思路:记录0元素所在行和列,然后清零
* time:o(mn)
* space:o(m+n)
*/
public static void setZero(int[][] matrix)