中级算法之数组和字符串：矩阵置零

矩阵置零

给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 0。请使用原地算法。

示例 1:

输入:

[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]

输出:

[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

示例 2:

输入:

[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]

输出:

[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

进阶:

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

因为…一开始看提示，没看到是使用额外空间，就想着原地改，遇到了无法存储坐标的问题，就很蠢…

思路1：使用一个矩阵记录0所在的位置（O（mn）的额外空间）

	public void setZeroes(int[][] matrix) {
		int temp[][] = new int[matrix.length][];
		// 拿出一个矩阵记录为0的位置
		for (int i = 0; i < matrix.length; i++) {
			temp[i] = new int[matrix[i].length];
			for (int j = 0; j < matrix[i].length; j++) {
				if (matrix[i][j] == 0)
					temp[i][j] = 1;
			}
		}
		for (int i = 0; i < temp.length; i++) {
			for (int j = 0; j < temp[i].length; j++) {
				if (temp[i][j] == 1) {
					for (int a = 0; a < temp.length; a++)
						matrix[a][j] = 0;
					for (int a = 0; a < temp[i].length; a++)
						matrix[i][a] = 0;
				}
			}
		}
	}

思路二：直接记录行和列即可（O（mn）的额外空间）

	public void setZeroes2(int[][] matrix) {
		int a[] = new int[matrix.length];// 记录哪一行
		int b[] = new int[matrix[0].length];// 记录哪一列
		for (int i = 0; i < matrix.length; i++) {
			for (int j = 0; j < matrix[i].length; j++) {
				if (matrix[i][j] == 0) {
					a[i] = 1;
					b[j] = 1;
				}
			}
		}

		for (int i = 0; i < matrix.length; i++) {
			for (int j = 0; j < matrix[i].length; j++) {
				if (a[i] == 1 || b[j] == 1) {
					matrix[i][j] = 0;
				}
			}
		}
	}

思路3：先将所有的0映射到第一行与第一列上，然后通过第一行与第一列最终将所有该之列的行列置零，注意，最后还要检查一下第一行与第一列是否也需要置零（O（1）的额外空间）

	public void setZeroes3(int[][] matrix) {
		boolean firstCol = false;// 记录第一行有没有0
		boolean firstRow = false;// 记录第一列有没有0
		for (int i = 0; i < matrix.length; i++) {
			for (int j = 0; j < matrix[0].length; j++) {
				if (matrix[i][j] == 0) {
					matrix[i][0] = 0;// 映射到第一列，最后根据第一列的0把所有该列的置零
					matrix[0][j] = 0;// 映射到第一行，最后根据第一行的0把所有该行的置零
					if (i == 0)
						firstCol = true;// 第一行有0，最后单独处理，把第一行置零
					if (j == 0)
						firstRow = true;
				}

			}
		}
		// 根据第一行与第一列的0开始全局置零
		for (int i = 1; i < matrix.length; i++) {
			for (int j = 1; j < matrix[0].length; j++) {
				if (matrix[0][j] == 0 || matrix[i][0] == 0) {
					matrix[i][j] = 0;
				}
			}
		}
		// 判断第一行与第一列是否有0
		if (firstCol) {
			for (int i = 0; i < matrix[0].length; i++)
				matrix[0][i] = 0;
		}
		if (firstRow) {
			for (int i = 0; i < matrix.length; i++)
				matrix[i][0] = 0;
		}
	}