《Cracking the Coding Interview程式設計師面試金典》----清除行列

請編寫一個演算法，若N階方陣中某個元素為0，則將其所在的行與列清零。

給定一個N階方陣int[][](C++中為vector<vector>)mat和矩陣的階數n，請返回完成操作後的int[][]方陣(C++中為vector<vector>)，保證n小於等於300，矩陣中的元素為int範圍內。

[[1,2,3],[0,1,2],[0,0,1]]

返回：[[0,0,3],[0,0,0],[0,0,0]]

解題思路：一看到這個題目可能會想到遍歷整個矩陣，只要發現值為0，就將其所在行和與列全部清零。這個是個錯誤的思想，當清零的時候，0元素覆蓋了還沒有遍歷到的元素，所以只有陣列有一個零，最後就導致整個陣列全為0。為了裝逼，學到一個size_t，應該是等於unsigned int的

#include<iostream>
#include<string.h>
#include<vector>
using namespace std;
vector<vector<size_t> > clearZero(vector<vector<size_t> > mat, size_t n) {
        vector<size_t>rows;
    vector<size_t >clows;
    for(size_t i = 0; i < mat.size(); ++i)
    {
        for(size_t j = 0; j < mat[0].size(); ++j)
        {
            if(mat[i][j] == 0)//將元素為0的行和列儲存
            {
                rows.push_back(i);
                clows.push_back(j);
            }
        }
    }
    for(size_t i = 0; i < rows.size(); ++i)//處理行
    {
        int row = rows[i];
        for(size_t j = 0; j < mat[0].size(); ++j)
        {

            mat[row][j] = 0;
        }
    }

    for(size_t j = 0; j < clows.size(); ++j) //處理列
    {
        size_t clow = clows[j];
        for(size_t i = 0; i < mat.size(); ++i)
        {
            mat[i][clow] = 0;
        }
    }

    return mat;
    }
void printVector(vector<vector<size_t> > mat,size_t n)
{
     for(size_t i = 0; i < n; i++)
    {
        for(size_t j = 0;j < n; j++)
        {
            cout << mat[i][j] << " ";
        }
        cout << endl;
    }
}
int main()
{
    vector<vector<size_t> > n;
    vector<size_t> v;
    size_t m;
    size_t temp;
    while (cin >> m)
    {
        v.clear();
        for (size_t i = 0; i<m; i++)
        {
            v.clear();
            for (size_t j = 0; j < m; j++)
            {
                cin >> temp;
                v.push_back(temp);
            }
            n.push_back(v);
        }
        printVector(clearZero(n, m),m);
    }
    return 0;
}

不懂的可以加我的QQ群：261035036（IT程式設計師面試寶典

群） 歡迎你的到來哦，看了博文給點腳印唄，謝謝啦~~