LeetCode之Set Matrix Zeroes

最新推荐文章于 2024-08-16 09:55:14 发布

懒行者

最新推荐文章于 2024-08-16 09:55:14 发布

阅读量383

点赞数

分类专栏： Leetcode 文章标签： Leetcode

Leetcode 专栏收录该内容

168 篇文章

订阅专栏

本文介绍了一种高效的矩阵置零算法，旨在将包含0元素的行和列全部置零，同时保持空间复杂度尽可能低。文章提供了三种不同空间复杂度的解决方案，并详细解释了如何仅使用常数额外空间实现这一目标。

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

/*显然，O(mn)空间复杂度的方法很简单，直接记录每个元素是否要设置为0.
O(m+n)空间复杂度的方法也比较简单，用两个数组记录该行是否需要设置为0.
常数空间复杂度方法需要利用matrix自己的空间：用matrix的第一行记录每列
是否需要设置为0，第一列记录每行是否需要设置为0.
方法参考自：http://www.cnblogs.com/TenosDoIt/p/3465273.html*/
class Solution {
public:
    void setZeroes(vector<vector<int> > &matrix) {
        bool is_row(false), is_col(false);//记录第一行、列是否需要设为0
        for(int i = 0; i < matrix[0].size(); ++i){
            if(matrix[0][i] == 0){
                is_row = true;
                break;
            }
        }
        for(int i = 0; i < matrix.size(); ++i){
            if(matrix[i][0] == 0){
                is_col = true;
                break;
            }
        }
        
        for(int i = 0; i < matrix.size(); ++i){
            for(int j = 0; j < matrix[0].size(); ++j){
                if(matrix[i][j] == 0){
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        
        for(int i = 1; i < matrix.size(); ++i){
            for(int j = 1; j < matrix[0].size(); ++j){
                if(matrix[i][0] == 0 || matrix[0][j] == 0) matrix[i][j] = 0;
            }
        }
        
        if(is_row) fill(matrix[0].begin(), matrix[0].end(), 0);
        if(is_col){
            for(int i = 0; i < matrix.size(); ++i){
                matrix[i][0] = 0;
            }
        }
    }
};