【LeetCode】 73. Set Matrix Zeroes

问题描述

https://leetcode.com/problems/set-matrix-zeroes/#/description

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

给定一个矩阵，如果一个元素为0，则其所在行和所在列都置为0.

注意：

乍一看很好解决——碰到一个为零的元素，就把它所在行列的其它元素置0就好了嘛，但实际一执行起来，就会有问题。因为如果就在原矩形上置0的话，最后的结果是整个矩阵都为0了。如下矩阵：

0 0 0 5 4 3 1 4 0 1 1 4 1 2 1 3 0 0 1 1 

如果碰到matrix(0,0) = 0就把第一行和第一列都置为了0，然后的循环就会把所有的元素都置为0，得不出最终结果：

0 0 0 0 0 0 0 4 0 0 0 0 0 0 0 3 0 0 0 0

算法

不in-place，记录要置0的行列。
使用两个数组rows和cols，分别记录要置0的行、列，遍历一次矩阵，如果当前元素(i,j)为0，则rows[i]=true, cols[j]=true，遍历结束后再根据rows和cols将矩阵该置0的行、列置0。时间复杂度O(n²)，空间复杂度O(m+n)。

代码

        public void setZeroes(int[][] matrix) {            if(matrix==null || matrix.length ==0) return;            boolean[] rows = new boolean[matrix.length]; // 需要重置为0的行记为true，否则为false            boolean[] cols = new boolean[matrix[0].length]; // 需要重置为0的列记为true，否则为false            for(int i=0;i<matrix.length;i++) {                for(int j=0;j<matrix[i].length;j++) {                    if(matrix[i][j]==0) {                        rows[i] = true;                        cols[j] = true;                    }                }            }            for(int i=0;i<rows.length;i++) {                if(rows[i])                    row2zeros(matrix, i);            }            for(int j=0;j<cols.length;j++) {                if(cols[j])                    col2zeros(matrix,j);            }        }        private void row2zeros(int[][] matrix, int row) {            for(int i=0;i<matrix[row].length;i++) {                matrix[row][i] = 0;            }        }        private void col2zeros(int[][] matrix, int col) {            for(int i=0;i<matrix.length;i++) {                matrix[i][col] = 0;            }        }