算法—清除矩阵中为0的行和列

题目

题目描述
请编写一个算法，若N阶方阵中某个元素为0，则将其所在的行与列清零。
给定一个N阶方阵int[]mat和矩阵的阶数n，请返回完成操作后的int[][]方阵(C++中为vector>)，保证n小于等于300，矩阵中的元素为int范围内。
测试样例：
[[1,2,3],[0,1,2],[0,0,1]]
返回：[[0,0,3],[0,0,0],[0,0,0]]

分析

可以将出现0的行列记录下来，设置两个boolean数组，一个记录行为0的情况，一个记录列为0的情况。
然后再遍历一次数组，如果当前行或者列出现过0，则将当前数清零。

代码

import java.util.*;

public class Clearer {
    public int[][] clearZero(int[][] mat, int n) {
        //可以将出现0的行列记录下来，设置两个boolean数组，一个记录行为0的情况，一个记录列为0的情况
        boolean[] rowFlag = new boolean[n];
        boolean[] columnFlag = new boolean[n];
        for(int i=0; i<n; i++){
            for(int j=0; j<n; j++){
                if(mat[i][j] == 0){
                    rowFlag[i] = true;
                    columnFlag[j] = true;
                }
            }
        }
        for(int i=0; i<n; i++){
            for(int j=0; j<n; j++){
                if(rowFlag[i] || columnFlag[j]){
                    mat[i][j] = 0;
                }
            }
        }
        return mat;
    }
}