【矩阵】清除行列

题目描述：

请编写一个算法，若N阶方阵中某个元素为0，则将其所在的行与列清零。
给定一个N阶方阵int[][](C++中为vector>)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]]

我的代码：

public int[][] clearZero(int[][] mat, int n) {        boolean row[] = new boolean[n];        boolean col[] = new boolean[n];                //记录为0的点的横纵坐标        for(int i = 0;i<n;i++){        for(int j = 0;j<n;j++){        if(mat[i][j]==0){        row[i] = true;        col[j] = true;        }        }        }                //重新生成数组        for(int i = 0;i<n;i++){        for(int j = 0;j<n;j++){        if(row[i] == true || col[j] == true){        mat[i][j] = 0;        }        }        }        return mat;    }