清除行列

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

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

关键：判断某个元素是否一开始就为0

public int[][] clearZero(int[][] mat, int n) {        // write code here        int[][] flag = new int[n][n];for (int i = 0; i < n; i++) {for (int j = 0; j < n; j++) {if (mat[i][j] == 0) {flag[i][j] = 1;}}}for (int i = 0; i < n; i++) {for (int j = 0; j < n; j++) {if (mat[i][j] == 0 && flag[i][j] == 1) {for (int k = 0; k < n; k++) {mat[i][k] = 0;mat[k][j] = 0;}}}}return mat;    }