清除行列

请编写一个算法，若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; }