LeetCode——Set Matrix Zeroes

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.

click to show follow up.

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?

思路：

1. 如果使用O(m+n)的空间，是比较好实现的，可以用两个数组分别记录各行和各列是否有零，然后再按照要求变化原来的矩阵。
2. 如果要求使用常量的空间，这个比较巧妙，我们可以使用原来矩阵的第一行和第一列来保存数据，而用两个标识符来保存初始时第一行和第一列是否有零，这样就可以巧妙的解决此问题。 
3. 具体步骤为：

a、先确定第一行和第一列是否需要清零。即，看看第一行中是否有0，记下来。也同时记下来第一列中有没有0。

b、扫描剩下的矩阵元素，如果遇到了0，就将对应的第一行和第一列上的元素赋值为0。这里不用担心会将本来第一行或第一列的1改成了0，因为这些值最后注定要成为0的。

c、根据第一行和第一列的信息，已经可以将剩下的矩阵元素赋值为结果所需的值了。即，拿第一行为例，如果扫描到一个0，就将这一列都清0.

d、根据a中确定的状态，处理第一行和第一列。如果最开始得到的第一行中有0的话，就整行清零。同理对列进行处理。

代码：

class Solution {public:void setZeroes(vector<vector<int> > &matrix) {int row = matrix.size();if (row == 0)return;int col = matrix[0].size();bool col_flag = false, row_flag = false;for (int i = 0; i < col; i++) {if (matrix[0][i] == 0) {col_flag = true;break;}}for (int i = 0; i < row; i++) {if (matrix[i][0] == 0) {row_flag = true;break;}}for (int i = 1; i < row; i++) {for (int j = 1; j < col; j++) {if (matrix[i][j] == 0) {matrix[0][j] = 0;matrix[i][0] = 0;}}}for (int i = 1; i < row; i++) {if (matrix[i][0] == 0) {for (int j = 1; j < col; j++) {matrix[i][j] = 0;}}}for (int i = 1; i < col; i++) {if (matrix[0][i] == 0) {for (int j = 1; j < row; j++) {matrix[j][i] = 0;}}}if (col_flag) {for (int i = 0; i < col; i++) {matrix[0][i] = 0;}}if (row_flag) {for (int i = 0; i < row; i++) {matrix[i][0] = 0;}}return;}};