tianya_team

欢迎订阅公众号:IT程序员开发

面试题91:清除矩阵0所在行列

题目:

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

给定一个MxN的int[][]矩阵(C++中为vector<vector>)mat和矩阵的阶数n,请返回完成操作后的int[][]矩阵(C++中为vector<vector>),保证n小于等于300,矩阵中的元素为int范围内。

思路:

定义两个bool类型的数组,分别表示矩阵的行或列是否出现0。

第一步:扫描整个矩阵,如果检测到某个元素为0,则将该元素所在行和列对应bool数组置1。

第二步:扫描bool数组,然后将该元素对应的行或列的元素全部置为0。

实现:

class Clearer {
public:
    vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {
        if(n==1) return mat;
        vector<bool> hang(n,0);
        vector<bool> lie(n,0);
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(mat[i][j]==0){
                    hang[i]=1;
                    lie[j]=1;
                }
            }
        }
        for(int i=0;i<n;i++){
            if(hang[i]==1){
                for(int j=0;j<n;j++){
                    mat[i][j]=0;
                }
            }
            if(lie[i]==1){
                for(int j=0;j<n;j++){
                    mat[j][i]=0;
                }
            }
        }
        return mat;
    }
};


阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/tianya_team/article/details/51563838
个人分类: algorithm
想对作者说点什么? 我来说一句

四维矩阵、四维数组

wl101yjx wl101yjx

2015-11-14 12:10:39

阅读数:5879

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭