程序员面试金典-7.清除行列

一、题目描述

链接：https://www.nowcoder.com/profile/5266727/codeBookDetail?submissionId=9096451
来源：牛客网

请编写一个算法，若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，就将其所在的行和列全部清零，这是错误的想法，因为清零的时候，覆盖掉了还没有遍历到的元素。因此，觉得总体思路是先对矩阵做遍历，记录出现0元素的横纵坐标，然后再对相应的行列元素置零。

三、代码

import java.util.*;
 
public class Clearer {
    public int[][] clearZero(int[][] mat, int n) {
        // write code here
        if(mat==null)
            return mat;
        HashSet<Integer> setx=new HashSet<Integer>(n);///或者使用两个boolean类型的数组来代替HashSet
        HashSet<Integer> sety=new HashSet<Integer>(n);
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(mat[i][j]==0){
                   sety.add(j);
                   setx.add(i);
                }
                 
                //break;
            }
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(setx.contains(i) || sety.contains(j))
                    mat[i][j]=0;
            }
        }
        return mat;
    }
}