Java 每日一题 6 附答案

大家好，我是青空。

快来和青空看看今天的Java每日一题吧

题目

编写一个算法，若M×N矩阵中某个元素为 0，则将其所在的行与列清零。

解法预警

不建议直接看下面答案，可以自己先思考一下怎么处理再看大看比较合适

解法

乍一看，这个问题似乎很简单:直接遍历整个矩阵，只要发现值为零的元素，就将其所在的 8 行与列清零。不过这种方法有个陷阱:在读取被清零的行或列时，读到的尽是零，于是所在行与列都得变成零。很快，整个矩阵的所有元素都会变为零。

避开这个陷阱的方法之一，是新建一个矩阵标记零元素位置。然后，在第二次遍历矩阵时将零元素所在行与列清零。这种做法的空间复杂度为O(MN)。

真的需要占用O(MN)空间吗?不是的。既然打算将整行和整列清为零，因此并不需要准确记录它是cell[2][4](行2、列4)，只需知道行2有个元素为零，列4有个元素为零。不管怎样，整行和整列都要清为零，又何必要记录零元素的确切位置?

下面是这个算法的实现代码。这里用两个数组分别记录包含零元素的所有行和列。然后，在第二次遍历矩阵时，若所在行或列标记为零，则将元素清为零。

public class Question {
	public static void nullifyRow(int[][] matrix, int row) {
		for (int j = 0; j < matrix[0].length; j++) {
			matrix[row][j] = 0;
		}		
	}

	public static void nullifyColumn(int[][] matrix, int col) {
		for (int i = 0; i < matrix.length; i++) {
			matrix[i][col] = 0;
		}		
	}		
	
	public static void setZeros(int[][] matrix) {
		boolean[] row = new boolean[matrix.length];	
		boolean[] column = 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; 
					column[j] = true;
		 		}
			}
		}
		
		for (int i = 0; i < row.length; i++) {
			if (row[i]) {
				nullifyRow(matrix, i);
			}
		}
		
		for (int j = 0; j < column.length; j++) {
			if (column[j]) {
				nullifyColumn(matrix, j);
			}
		}
	}	
}