【程序员面试金典】清除行列

【程序员面试金典】清除行列

题目

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

分析

边遍历边清空可能产生连锁反应，即将应该被清空且不为0的元素提前清空，导致该元素同行同列所有元素被清空。所以需要在遍历时记录需要清空的行列。

解答

    // 按点来记录，适合稠密矩阵
    vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {
        // write code here
        vector<int> tmp;
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < n; j++)
            {
                if (mat[i][j] == 0) tmp.push_back(i * n + j);
            }
        }
        for (unsigned i = 0; i < tmp.size(); i++)
        {
            int row = tmp[i] / n;
            int col = tmp[i] % n;
            for (int k = 0; k < n; k++)
            {
                mat[row][k] = 0;
                mat[k][col] = 0;
            }
        }
        return mat;
    }

    // 按行列记录，适合稀疏矩阵
    vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {
        // write code here
        set<int> rowTmp, colTmp;
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < n; j++)
            {
                if (mat[i][j] == 0){
                    rowTmp.insert(i);
                    colTmp.insert(j);
                }
            }
        }
        auto begin1 = rowTmp.begin(), begin2 = colTmp.begin();
        while (begin1 != rowTmp.end()){
            for (int k = 0; k < n; k++) mat[*begin1][k] = 0;
            begin1++;
        }
        while (begin2 != colTmp.end()){
            for (int k = 0; k < n; k++) mat[k][*begin2] = 0;
            begin2++;
        }
        return mat;
    }