73. Set Matrix Zeroes

Given an m x n integer matrix matrix, if an element is 0, set its entire row and column to 0's, and return the matrix.

You must do it in place.

class Solution {

    public void setZeroes(int[][] matrix) {

        List<Integer> row = new ArrayList();

        List<Integer> col = new ArrayList();

        int n = matrix.length;

        int m = matrix[0].length;

        for(int i = 0; i < n; i++) {

            for(int j = 0; j < m; j++) {

                if(matrix[i][j] == 0) {

                    row.add(i);

                    col.add(j);

                }

            }

        }

        for(int i = 0; i < n; i++) {

            for(int j = 0; j < m; j++) {

                if(row.contains(i) || col.contains(j)) {

                    matrix[i][j] = 0;

                }

            }

        }

    }

}

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容