面试题 01.08. 零矩阵

面试题 01.08. 零矩阵

编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。

 

示例 1：

输入：
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
输出：
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

思路

增加一个辅助空间

class Solution {
private:
	void change(vector<vector<int>>& matrix, int row, int col){
		for(int i=0;i<matrix.size();i++) matrix[i][col]=0;
		for(int i=0;i<matrix[row].size();i++) matrix[row][i]=0;
	}
public:
    void setZeroes(vector<vector<int>>& matrix) {
        vector<vector<int> > copy = matrix;
        for(int i=0;i<copy.size();i++){
        	for(int j=0;j<copy[i].size();j++){
        		if(copy[i][j]==0)
        			change(matrix, i, j);
			}
		}
    }
};