【LeetCode-Java实现】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;
                }
            }
        }        
    }

2. 借助额外空间：两个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;
                }
            }
        }    
    }