[程序员面试宝典]清除行列

yao_yao_2015

于 2017-04-13 15:57:56 发布

阅读量422

点赞数

CC 4.0 BY-SA版权

分类专栏：程序员面试宝典

本文链接：https://blog.youkuaiyun.com/yao_yao_2015/article/details/70158822

程序员面试宝典专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种高效的矩阵处理算法——矩阵置零算法。该算法通过记录需要置零的行列索引来快速将矩阵中指定元素所在行和列的所有元素置零。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：

设置一个行的矩阵，和列的矩阵，保存元素为0的行号和列号

如果行号或者列号被标记，赋值为0

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