【LeetCode-Java实现】73. Set Matrix Zeroes矩阵置0

73. Set Matrix Zeroes矩阵置0

题目描述

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]
]
找出矩阵中的0,把0所在的行和列全都置为0

思路

  1. 直接法
    遍历找到矩阵中所有的0,把0所在的行和列所有元素设置为特定值,然后再次遍历矩阵,将被设为特定值的元素设为0
  2. 借助集合set存放找到的0元素所在的行号和列号
    然后利用集合查询元素的方法,遍历矩阵,若找到了集合中的元素,就设为0

实现

  1. 直接法 :时间复杂度O(mn(m+n)) 空间复杂度O(1)
public void setZeroes(int[][] matrix) {
        //直接遍历找到符合条件的,再将行列置为0
        int R=matrix.length;     //行数
        int C=matrix[0].length;  //列数
        
        //定义一个很大的负虚数(确保不和数组中已有元素重复)
        int MODIFIED=-1000000;
            
        //遍历找0
        for(int i=0;i<R;i++){
            for(int j=0;j<C;j++){
                if(matrix[i][j]==0){
                    //将第i行,第j列中不为0的元素设为很大的负虚数MODIFIED
                    for(int k=0;k<C;k++){
                        if(matrix[i][k]!=0){
                            matrix[i][k]=MODIFIED;
                        }
                    }
                    for(int k=0;k<R;k++){
                        if(matrix[k][j]!=0){
                            matrix[k][j]=MODIFIED;
                        }
                    }
                }
            }
        }
        //将被设置为MODIFIED的元素设为0
        for(int i=0;i<R;i++){
            for(int j=0;j<C;j++){
                if(matrix[i][j]==MODIFIED){
                    matrix[i][j]=0;
                }
            }
        }        
    }
  1. 借助额外空间:两个set分别保存0元素所在的行号和列号 时间复杂度O(m*n) 空间复杂度O(m+n)
public void setZeroes(int[][] matrix) {
        //额外空间,借助两个集合set存放0元素所在的行号和列号
        int R=matrix.length;     //行数
        int C=matrix[0].length;  //列数
        
        Set<Integer> rows=new HashSet<>();//存放0元素行号的集合
        Set<Integer> cols=new HashSet<>();//存放0元素列号的集合
        
        for(int i=0;i<R;i++){
            for(int j=0;j<C;j++){
                if(matrix[i][j]==0){
                    rows.add(i);   //把0元素所在的所有行号加到rows集合中
                    cols.add(j);   //把0元素所在的所有列号加到cols集合中
                }
            }
        }
        for(int i=0;i<R;i++){
            for(int j=0;j<C;j++){
                if(rows.contains(i) || cols.contains(j)){   //若i或j是集合中的元素,就置0
                    matrix[i][j]=0;
                }
            }
        }    
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值