首页 > 代码库 > Leetcode: Set Matrix Zeroes
Leetcode: Set Matrix Zeroes
一次过,空间复杂度为O(m+n), 下一次做的时候寻找constant space solution。用boolean array也可以,用bit vector可能会更节省
1 import java.util.*; 2 3 public class Solution { 4 public void setZeroes(int[][] matrix) { 5 HashSet<Integer> rows = new HashSet<Integer>(); 6 HashSet<Integer> columns = new HashSet<Integer>(); 7 8 for (int i = 0; i < matrix.length; i++) { 9 for (int j = 0; j < matrix[0].length; j++) { 10 if (matrix[i][j] == 0) { 11 rows.add(i); 12 columns.add(j); 13 } 14 } 15 } 16 17 for (int i = 0; i < matrix.length; i++){ 18 for (int j = 0; j < matrix[0].length; j++) { 19 if (rows.contains(i) || columns.contains(j)) matrix[i][j] = 0; 20 } 21 } 22 } 23 }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。