leetcode-73 Set Matrix Zeroes

最新推荐文章于 2024-08-16 09:55:14 发布

原创最新推荐文章于 2024-08-16 09:55:14 发布 · 93 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

100 篇文章

订阅专栏

本文介绍了一种高效的算法，用于处理矩阵中元素为零的情况，将其所在行和列的所有元素置为零，实现方式是在原地操作，避免了额外的空间消耗。通过使用两个集合分别记录含有零元素的行和列，再遍历矩阵进行置零操作。

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

Example 1:

Input: 
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
Output: 
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

Example 2:

Input: 
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
Output: 
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

Follow up:

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?

最简单的方法找出所有0的位置，然后对其行和列变为0

public void setZeroes(int[][] matrix) {
   Set<Integer> listI = new HashSet<Integer>();
   Set<Integer> listJ = new HashSet<Integer>();
for(int i = 0;i<matrix.length;i++) {
   for(int j=0;j<matrix[i].length;j++) {
       if(matrix[i][j]==0) {
           listI.add(i);
           listJ.add(j);
       }
   }
}

for(int i = 0;i<matrix.length;i++) {
   for(int j=0;j<matrix[i].length;j++) {
       if(listI.contains(i) || listJ.contains(j)) {
           matrix[i][j]=0;
       }
   }
}
}