Leetcode73：矩阵置零

题目描述

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

代码实现1

	/**
	 * 对于这个矩阵中，只要元素为0，那么他所在的行的第一个元素和列的一个元素一定是0 定义row和col两个布尔变量，为了记录第一行和第一列是否需要遍历为0
	 * 
	 * @return
	 */
	public static void setZeroes(int[][] matrix) {
		if (matrix == null || matrix.length == 0 || matrix[0].length == 0) return;
		int m = matrix.length;
		int n = matrix[0].length;
		boolean row = false;
		boolean col = false;
		for (int i = 0; i < m; i++) {
			for (int j = 0; j < n; j++) {
				if (matrix[i][j] == 0) {// 如果这个元素为0
					matrix[i][0] = 0;// 将这一行的第一个元素置为0
					matrix[0][j] = 0;// 将这一列的第一个元素置为0
					if (i == 0)
						row = true;// 如果0在第一行,row定为true
					if (j == 0)
						col = true;// 如果0在第一列，col定为true
				}
			}
		}
		// 按行遍历，从第二行开始
		for (int i = 1; i < m; i++) {
			if (matrix[i][0] == 0) {
				for (int j = 1; j < n; j++) {// 如果第i行的第一个元素为0，那么将这一行的元素都置为0
					matrix[i][j] = 0;
				}
			}
		}
		// 按列遍历，从第二列开始
		for (int j = 1; j < n; j++) {
			if (matrix[0][j] == 0) {
				for (int i = 1; i < m; i++) {
					matrix[i][j] = 0;
				}
			}

		}
		if (row) {// 如果原矩阵第一行有0，遍历第一行为0
			for (int j = 0; j < n; j++) {
				matrix[0][j] = 0;
			}
		}
		if (col) {// 如果原矩阵第一列有0，遍历第一列为0
			for (int i = 0; i < m; i++) {
				matrix[i][0] = 0;
			}
		}
	}

代码实现2

	/**
	  	我们扫描一遍原始矩阵，找到所有为零的元素。
		如果我们找到 [i, j] 的元素值为零，我们需要记录下行号 i 和列号 j。
		用两个 sets ，一个记录行信息一个记录列信息。
	 */
public void setZeroes(int[][] matrix) {
		int R = matrix.length;
		int C = matrix[0].length;
		Set<Integer> rows = new HashSet<Integer>();
		Set<Integer> cols = new HashSet<Integer>();

		for (int i = 0; i < R; i++) {
			for (int j = 0; j < C; j++) {
				if (matrix[i][j] == 0) {
					rows.add(i);
					cols.add(j);
				}
			}
		}

		for (int i = 0; i < R; i++) {
			for (int j = 0; j < C; j++) {
				if (rows.contains(i)||cols.contains(j)) {
					matrix[i][j]=0;
				}
			}

		}

	}