清除行列

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