[LeetCode]73. Set Matrix Zeroes

73. Set Matrix Zeroes

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

Follow up:
Did you use extra space?
A straight forward solution using O(mn) space is probably a bad idea.
A simple improvement uses O(m + n) space, but still not the best solution.
Could you devise a constant space solution?

分析

题目要求尽量小的空间使用，其中O(m+n)可以通过使用两个数组来记录每行每列的是否存在0；然后再遍历各行各列。

源码

    void setZeroes(vector<vector<int>>& matrix) {
        int rows = matrix.size();
        int cols = matrix[0].size();
        vector<bool> rzeros(rows,false), czeros(cols,false);
        for(int r = 0; r < rows; r++) {
            for(int c = 0; c < cols; c++) {
                if(matrix[r][c] == 0) {
                    rzeros[r] = true;
                    czeros[c] = true;
                }
            }
        }
        for(int r = 0; r < rows; r++) {
            if(rzeros[r]) {
                for(int c = 0; c < cols; c++) {
                    matrix[r][c] = 0;
                }
            }
        }
        for(int c = 0; c < cols; c++) {
            if(czeros[c]) {
                for(int r = 0; r < rows; r++) {
                    matrix[r][c] = 0;
                }
            }
        }

    }