清除行列（程序员面试金典）

清除行列

• 参与人数：1233时间限制：3秒空间限制：32768K
• 通过比例：27.49%
• 最佳记录：20 ms|8700K（来自  OOACMer）
•  算法知识视频讲解

题目描述

请编写一个算法，若MxN矩阵中某个元素为0，则将其所在的行与列清零。

给定一个MxN的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]]

链接：http://www.nowcoder.com/practice/c95aac1506574dfc8ad44c3939c6739d?rp=1&ru=/ta/cracking-the-coding-interview&qru=/ta/cracking-the-coding-interview/question-ranking

思路：没想到比O(m*n)时间复杂度还快的，有思路的可以交流下；

没什么别的方法，我用row，line来记录行列中是否有0；首先初始化为1；当mat[i][j]出现0的时候，将row[i]=0;line[j]=0；  遍历一遍；

最后，再遍历一遍把row或者line里有0 的地方就把值变成0，否则记录原值！

方法比较直接。。。

/**
题目描述

请编写一个算法，若MxN矩阵中某个元素为0，则将其所在的行与列清零。
给定一个MxN的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]]
**/

#include<iostream>
#include<vector>
using namespace std;

class Clearer {
public:
    vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {
        vector<vector<int> > vec;
        if(mat.size()<=0||n==0) return vec;

        vector<int> row(mat.size(),1);
        vector<int> line(n,1);
        for(int i=0;i<mat.size();i++)
        {
            for(int j=0;j<mat[i].size();j++)
            {
                if(!mat[i][j])
                {
                    row[i]=0;
                    line[j]=0;
                }
            }
        }

        for(int i=0;i<mat.size();i++)
        {
            vector<int> tmp;
            for(int j=0;j<mat[i].size();j++)
            {
                if(!row[i] || !line[j])
                    tmp.push_back(0);
                else tmp.push_back(mat[i][j]);
            }
            vec.push_back(tmp);
            tmp.clear();
        }
        return vec;
    }
};
void Print(vector<vector<int> > &v)
{
    for(size_t i=0;i<v.size();i++)
    {
        for(size_t j=0;j<v[i].size();j++)
            cout<< v[i][j] <<" ";
        cout<<endl;
    }
}
int main()
{
    vector<vector<int> >arr={{1,2,3},{0,1,2},{0,0,1}};
    Print(arr);
    Clearer cl ;
    arr=cl.clearZero(arr,3);
    Print(arr);
    return 0;
}