清除行列

题目：请编写一个算法，若MxN矩阵中某个元素为0，则将其所在的行与列清零。给定一个MxN的int[][]矩阵(C++中为vector>)mat和矩阵的阶数n，请返回完成操作后的int[][]矩阵(C++中为vector>)，保证n小于等于300，矩阵中的元素为int范围内

思路：初次看这道题目非常简单，直接遍历该MN矩阵，如果遇到0,就将其所在的行和列清0，但这样，被清0和原始的0无法区分，导致全部被清0的问题。所有容易想到，先申请一个新的MN矩阵，存储原MN的矩阵，将新的MN矩阵作为标记，扫描新的MN矩阵，遇到0，将原矩阵对应的行和列清0.这样将占用MN的存储空间，其实只需申请一个M大小和N大小的向量存储有0的行和列即可，实现如下。

vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {
        // write code here
        vector<int> row(n,0),col(n,0);
        for (int i = 0; i < n; i++){
            for (int j = 0; j < n; j++){
                if (mat[i][j] == 0){
                    row[i] = 1;
                    col[j] = 1;
                }
            }
        }

    //若行i或列j有元素为0，则将mat[i][j]设置为0
        for (int i = 0; i < n; i++){
            for (int j = 0; j < n; j++){
                if (row[i] || col[j]){
                    mat[i][j] = 0;
                }
            }
        }
        return mat;
    }