程序员面试金典1.7数组 清除行列

题目描述
请编写一个算法，若N阶方阵中某个元素为0，则将其所在的行与列清零。

给定一个N阶方阵int[]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]]
遍历整个矩阵，记录出现0位置的行号和列号，然后分别对行和列置0即可，代码如下

 vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {
        // write code here
        vector<int> row;//保存出现0位置的行号
        vector<int> col;//保存出现0位置的列号
        for(int i = 0;i<n;++i)
        {
            for(int j = 0;j<n;++j)
            {
                if(0==mat[i][j])
                {
                    row.push_back(i);
                    col.push_back(j);
                }
            }
        }
        for(int i = 0;i<row.size();i++)//对行置0
        {
            for(int j = 0;j<n;++j)
            {
                mat[row[i]][j]=0;
            }
        }
        for(int i = 0;i<col.size();++i)//对列置0 
        {
            for(int j = 0;j<n;j++)
            {
                mat[j][col[i]]=0;
            }
        }
        return mat;
    }