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;
	}
};