在现实生活中,我们常常会遇到一些情况:一个系统中的某个部分出问题了,这个问题可能会影响整个系统的运行。今天,我们就要解决一个类似的问题:矩阵置零。想象一下,一个矩阵中某个元素变成了零,这就像你的电脑某个硬件坏了,可能会影响其他硬件的正常运行。我们要做的就是将受影响的部分(行和列)都置为零。
题目描述
给定一个 m x n
的矩阵,如果一个元素是 0
,则将其所在的行和列的所有元素都设为 0
。请使用原地算法。
例如:
- 输入:
[[1,1,1],[1,0,1],[1,1,1]]
- 输出:
[[1,0,1],[0,0,0],[1,0,1]]
解题思路
我们可以考虑几种不同的解法来解决这个问题,从最简单直接的方法到优化的空间复杂度方法。
方法一:使用额外空间
思路
最直观的方法是使用两个数组来记录矩阵中哪些行和哪些列需要置零。我们遍历整个矩阵,一旦遇到元素为 0
,就将对应的行和列标记为需要置零。然后我们再次遍历矩阵,将需要置零的行和列置零。
代码实现
public class