LeetCode73、矩阵置零

题目描述

        给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。（原地算法：是一种使用小的，固定数量的额外之空间来转换资料的算法。）

解题思路

        本题的关键在于标记矩阵中0元素所在的行和列，最简单的方法是使用两个标记数组，此时的算法空间复杂度O(m+n)，为了降低空间复杂度，我们使用原数组的第一列和第一行作为0元素标记位，但这会导致原数组的第一列和第一行被修改，无法记录原来是否含有0。我们可以用一个标记变量来记录第一列是否有0，.同时可以倒序遍历数组以防止提前修改第一行的元素。这样一来算法的空间复杂度位O(1)。

具体步骤是：

第一步：第一次正序遍历数组，标记矩阵每一行、每一列是否有0出现。如果第一列有零，标记变量为真。如果该元素为0，则将其所在行、所在列的第一个元素置零，用于标记0元素的位置

第二步：第二次倒叙遍历数组，更新数组元素。如果该元素所在行或所在列的第一个元素为0，则将该元素置零。最后如果第一列有零元素，则将第一列元素置零。

java代码实现

public void setZeroes(int[][] matrix) {
		int m=matrix.length;
		int n=matrix[0].length; 
		//标记第一列是否有0
		boolean tagCol=false;
		//标记值为0的元素所在的行和列
		for (int i = 0; i < m; i++) {
			//标记第一列是否有0
			if (matrix[i][0]==0) {
				tagCol=true;
			}
			//如果matrix[i][j]==0，则将给元素所在行和列的第一个元素设置为0
			for (int j = 1; j < n; j++) {
				if (matrix[i][j]==0) {
					matrix[i][0]=matrix[0][j]=0;
				}
			}
		}
		//矩阵元素置0
		for (int i = m-1; i >=0; i--) {
			
			for (int j = n-1; j >0; j--) {
				if (matrix[i][0]==0||matrix[0][j]==0) {
					matrix[i][j]=0;
				}
			}
			//第一列元素置零
			if (tagCol) {
				matrix[i][0]=0;
			}
		}        
    }