清除行列

题目描述

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

class Clearer {
public:
    vector<vector<int>> clearZero(vector<vector<int>> mat, int n) {
        // write code here
        //vector<vector<int>> res;
        if(mat.size() == 0)
            return mat;

        int rowSize = mat.size();
        int colSize = mat[0].size();

        bool row[300] = {0};
        bool col[300] = {0};

        for(int i = 0;i < rowSize; i++){
            for(int j = 0;j < colSize; j++){
                if(mat[i][j] == 0){
                    row[i] = true;
                    col[j] = true;
                }
            }
        }

        for(int i = 0;i < rowSize;i++){
            for(int j = 0;j < colSize;j++){
                if(row[i] || col[j]){
                    mat[i][j] = 0;
                }
            }
        }

        return mat;
    }
};