力扣73.矩阵置零(java语言)

题目描述：

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

解题思路及代码：

思路1：

我们容易想到一个比较直接的方法：指标法，我们定义两个辅助布尔型数组，用于记录矩阵行列为0的位置。我们遍历矩阵，若遍历到为0的元素时，让两个辅助数组分别对其行、列坐标做标志(标志为true)，再次遍历矩阵，若辅助数组中的其中一个位置为true则将该位置置为0(语言描述可能不太清楚，也可以直接看代码，代码比较容易理解)

代码1：

class Solution {
	//Tiem Complexity :O(MN)
	//Space Complexity: O(M + N)
	 public void setZeroes(int[][] matrix) {
	 	//定义辅助行、列标记数组
	 	boolean[] row = new boolean[matrix.length];
	 	boolean[] col = new boolean[matrix[0].length];
	 	//遍历矩阵，标记辅助数组
	 	for (int i = 0; i < matrix.length; i++) {
	 		for (int j = 0; j < matrix[0].length; j++) {
	 			if (matrix[i][j] == 0) {
	 				//标记
	 				row[i] = true;
	 				col[j] = true;
	 			}
	 		}
	 	}
	 	//再次遍历并置为0
	 	for (int i = 0; i < matrix.length; i++) {
	 		for (int j = 0; j < matrix[0]; j++) {
	 			if (row[i] = true || col[j] = true) {
	 				matrix[i][j] = 0;
	 			}
	 		}
	 	}
	 }
}

思路2：

思路1我们使用了额外的空间来标记置换0，但其实我们可以以矩阵的第一行和列作为标准，然后我们从矩阵的第二行和列开始遍历，当遍历到为0的元素时(假设此时该元素坐标为i，j)，将对应[0][j]和[i][0]置为0，再次从矩阵第二行列开始置换0(操作与思路1一致)，但是我们若以矩阵的第一行列为标志，我们也不能忘记考虑第一行列存在0的情况！！！为此我们定义两个布尔变量用于标记，并在最后对第一行列单独做一次处理(若第一行列存在0)

代码2：

class Solution  {
	//Time Complexity: O(MN)
	//Space Complexity: O(1)
	public void setZeroes(int[][] matrix) {
		int row = matrix.length;
        int col = matrix[0].length;
		//定义用于判断第一行和列的标志
		boolean row0_flag = false;
		boolean col0_flag = false;

		//判断第矩阵第一行、列是否本身存在0 
		for (int i = 0; i < col; i++) {
			if (matrix[0][i] == 0) {
				row0_flag = true;
				break;
			}
		}
		for (int j = 0; j < row; j++) {
			if (martix[j][0] == 0) {
				col0_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] = matrix[i][0] = 0;
				}
			} 
		}
		// 开始置换为0
		for (int i = 1; i < row; i++) {
			for (int j = 1; j < col; j++) {
				if (matrix[0][j] == 0 || matrix[i][0] == 0) {
					matrix[i][j] = 0;
				}
			}
		}
		//若矩阵第一行或者列存在0
        //则将第一行或列全置为0
         if (row0_flag) {
             for (int i = 0; i < col; i++) {
                  matrix[0][i] = 0;
               }
           }
           if (col0_flag) {
               for (int j = 0; j < row; j++) {
                   matrix[j][0] = 0;
               }
           }
	}
}