LeetCode73. 矩阵置零

给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 0。请使用原地算法。
关于原地算法，名字倒是挺响，https://baike.baidu.com/item/原地算法
示例 1:

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

进阶:

• 一个直接的解决方案是使用 O(mn) 的额外空间，但这并不是一个好的解决方案。
• 一个简单的改进方案是使用 O(m + n)的额外空间，但这仍然不是最好的解决方案。
• 你能想出一个常数空间的解决方案吗？

最开始想到的是：

class Solution {
	static class Pair {
		int i, j;

		public Pair(int i, int j) {
			this.i = i;
			this.j = j;
		}
	}

	public void setZeroes(int[][] matrix) {
		List<Pair> zeroes = new LinkedList<>();
		for (int i = 0; i < matrix.length; ++i) {
			for (int j = 0; j < matrix[i].length; ++j) {
				if (matrix[i][j] == 0) {
					zeroes.add(new Pair(i, j));
				}
			}
		}
		for (Pair p : zeroes) {
			setZero(p, matrix);
		}
	}

	public void setZero(Pair p, int[][] matrix) {
		for (int i = 0; i < matrix[p.i].length; ++i)
			matrix[p.i][i] = 0;
		for (int j = 0; j < matrix.length; ++j)
			matrix[j][p.j] = 0;
	}
}

进阶，可以常数空间解决
思路：
矩阵第一列和第一行每个值分别标记该行或者列是否需要置零。
那对于第一行第一列是否需要置零，用两个boolean标记就可以了

class Solution {
	public void setZeroes(int[][] matrix) {

		boolean firstColumn = false, firstLine = false;
		for (int i = 0; i < matrix.length; ++i) {
			if (matrix[i][0] == 0) {
				firstColumn = true;
				break;
			}
		}
		for (int i = 0; i < matrix[0].length; ++i) {
			if (matrix[0][i] == 0) {
				firstLine = true;
				break;
			}
		}
		for (int i = 1; i < matrix.length; ++i) {
			for (int j = 1; j < matrix[i].length; ++j) {
				if (matrix[i][j] == 0) {
					matrix[i][0] = 0;
					matrix[0][j] = 0;
				}
			}
		}
		for (int i = 1; i < matrix.length; ++i) {
			if (matrix[i][0] == 0) {
				for (int j = 0; j < matrix[i].length; ++j)
					matrix[i][j] = 0;
			}
		}
		for (int i = 1; i < matrix[0].length; ++i) {
			if (matrix[0][i] == 0) {
				for (int j = 0; j < matrix.length; ++j)
					matrix[j][i] = 0;
			}
		}
		if (firstLine) {
			for (int i = 0; i < matrix[0].length; ++i) {
				matrix[0][i] = 0;
			}
		}
		if (firstColumn) {
			for (int i = 0; i < matrix.length; ++i) {
				matrix[i][0] = 0;
			}
		}
	}
}