leecode 解题总结：73. Set Matrix Zeroes

#include <iostream>
#include <stdio.h>
#include <vector>
using namespace std;
/*
问题:
Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

click to show follow up.

分析:这是程序员面试金典的一道题目。元素为0，需要将该元素所在行和列都置为0。

输入:
3(行数) 4(列数)
1 1 0 1
1 0 1 1
1 1 1 1
输出:
0 0 0 0
0 0 0 0
1 0 0 1

关键:
1 注意如果如果将所在行和列设置为0,后，再遍历矩阵，会把整个矩阵变成0.
为了避免这种情况，当遍历到某个元素为0时，先记录这个元素的行号和列号，最后再调整
需要清零的行号存入行号数组rowArr，需要清零的列号存入列号数组colArr。
下一次再遍历某个元素是A[i][j]，如果rowArr[i] == 1 或者colArr[j] == 1,就设置为0
*/

class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
if(matrix.empty())
{
return ;
}
int row = matrix.size();
int col = matrix.at(0).size();
vector<int> rowArr(row ,  0);//初始化0表示非删除标记
vector<int> colArr(col , 0);
for(int i = 0 ; i < row ; i++)
{
for(int j = 0 ; j < col ;j++)
{
//遇到0元素，记录0元素的行号和列号
if(matrix.at(i).at(j) == 0)
{
rowArr.at(i) = 1;
colArr.at(j) = 1;
}
}
}
for(int i = 0 ; i < row ; i++)
{
for(int j = 0 ; j < col ;j++)
{
//如果当前元素的行号或者列号是对应0元素的标记，就设置
if( 1 == rowArr.at(i) || 1 == colArr.at(j))
{
matrix.at(i).at(j) = 0;
}
}
}
}
};

void print(vector<vector<int>>& result)
{
if(result.empty())
{
cout << "no result" << endl;
return;
}
int size = result.size();
int len = result.at(0).size();
for(int i = 0 ; i < size ; i++)
{
for(int j = 0 ; j < len ; j++)
{
cout << result.at(i).at(j) << " " ;
}
cout << endl;
}
}

void process()
{
int row;
int col;
Solution solution;
vector<vector<int>> matrix;
int value;
while(cin >> row >> col )
{
matrix.clear();
for(int i = 0 ; i < row ; i++)
{
vector<int> nums;
for(int j = 0 ; j < col ; j++)
{
cin >> value;
nums.push_back(value);
}
matrix.push_back(nums);
}
solution.setZeroes(matrix);
print(matrix);
}
}

int main(int argc , char* argv[])
{
process();
getchar();
return 0;
}