C++ 清除行列

#include <vector>
#include<iostream>
using namespace std;
class Clearer {
public:
    vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {
        // write code here
        int row_number[n];
        for(int i=0;i<n;i++){
            row_number[i]=0;
        }
        // C++是会给数组进行随机初始化的，故而在这里必须要逐个对于数组中的所有元素进行初始化
        int col_number[n];
        for(int i=0;i<n;i++){
            col_number[i]=0;
        }

        int start=0;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(mat[i][j]==0){
                    row_number[i]=1;
                    col_number[j]=1;
                }
            }
        }

        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(row_number[i]==1 or col_number[j]==1)
                    mat[i][j]=0;
            }
        }
        return mat;
    }
};


int main(){
    vector<vector<int> > vec,output;
    vector<int> a;
    a.push_back(1);
    a.push_back(2);
    a.push_back(3);

    vector<int> b;
    b.push_back(0);
    b.push_back(1);
    b.push_back(2);

    vector<int> c;
    c.push_back(0);
    c.push_back(0);
    c.push_back(1);

    vec.push_back(a);
    vec.push_back(b);
    vec.push_back(c);

    Clearer clr;
    output=clr.clearZero(vec,3);

    for(int i=0;i<3;i++){
        for(int j=0;j<3;j++){
            cout<<output[i][j]<<" ";
        }
        cout<<endl;
    }

    return 0;
}