程序员面试金典——清除行列

题目描述

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

给定一个N阶方阵int[][](C++中为vector<vector>)mat和矩阵的阶数n，请返回完成操作后的int[][]方阵(C++中为vector<vector>)，保证n小于等于300，矩阵中的元素为int范围内。

测试样例：

[[1,2,3],[0,1,2],[0,0,1]]

返回：[[0,0,3],[0,0,0],[0,0,0]]

解题思路：遍历整个数组，将为0的所在行和列分别加到row与col的hashset里面（这里也可以用数组）

然后依次判定行，然后判定列，hashset含有的就将该行，该列置为0。

import java.util.*;
import java.util.HashSet;
public class Clearer {
    public int[][] clearZero(int[][] mat, int n) {
        // write code here
        HashSet<Integer> col = new HashSet<>();
		HashSet<Integer> row = new HashSet<>();
		for(int i = 0 ;i<n;i++){
			for(int j = 0;j<n;j++){
				if(mat[i][j]==0){
					row.add(i);
					col.add(j);
				}
			}
		}
		
		for(int i = 0;i<n;i++){
			if(row.contains(i)){
				for(int j = 0;j<n;j++){
					mat[i][j]=0;
				}
			}
		}
		
		for(int j = 0;j<n;j++){
			if(col.contains(j)){
				for(int i = 0;i<n;i++){
					mat[i][j]=0;
				}
			}
		}
		
		return mat;
    }
}