清除行列

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

class Clearer {
public:
    vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {
        // write code here
       set<int> hang;
        set<int> lie;
        for (int i = 0; i<n; i++)
        {
            for (int j = 0; j<n; j++)
            {
                if (mat[i][j] == 0)
                {
                    hang.insert(i);
                    lie.insert(j);
                }
            }
        }
        set<int>::iterator phang;
        set<int>::iterator plie;
        phang = hang.begin();
        plie = lie.begin();
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < n; j++)
            {
                if (i == (*phang) || j == (*plie))
                    mat[i][j] = 0;
                if (j == (*plie))
                {
                    plie++;
                    if (plie == lie.end())
                        plie--;
                }
            }
            plie = lie.begin();
            if (i == (*phang))
                phang++;
        }
        return mat;
    }
};