leetcode【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.

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]
]

源码：

用两个set存着曾今为0的行列。方法非常低效。空间还可以，毕竟没有创建啥新的数组。

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        set<int> rows, cols;
        int m = matrix.size(), n = matrix[0].size();
        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                if(matrix[i][j] == 0){
                    rows.insert(i);    cols.insert(j);
                }
            }
        }
        for(std::set<int>::iterator it=rows.begin(); it!=rows.end(); ++it){
            for(int j=0; j<n; j++)      matrix[*it][j] = 0;
        }
        for(std::set<int>::iterator it=cols.begin(); it!=cols.end(); ++it){
            for(int i=0; i<m; i++)      matrix[i][*it] = 0;
        }
    }
};

leetcode官方给的高效解法，把所有的映射左边和上边，遍历的时候考虑中心的。用一个c存储左侧情况，用matrix[0][0]存储上侧情况。貌似没有我的方法高效；

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m=matrix.size(), n=matrix[0].size();
        bool c = false;
        for(int i=0; i<m; i++){
            if(!matrix[i][0])   c = true;
            for(int j=1; j<n; j++){
                if(!matrix[i][j])   matrix[i][0] = matrix[0][j] = 0;
            }
        }
        for(int i=1; i<m; i++){
            for(int j=1; j<n; j++){
                if(!matrix[i][0] || !matrix[0][j])  matrix[i][j] = 0;
            }
        }
        if(!matrix[0][0]){
            for(int j=0; j<n; j++)      matrix[0][j] = 0;
        }
        if(c){
            for(int i=0; i<m; i++)      matrix[i][0] = 0;
        }
    }
};