73. Set Matrix Zeroes

最新推荐文章于 2022-10-28 21:42:54 发布

6plus

最新推荐文章于 2022-10-28 21:42:54 发布

阅读量220

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/gcola007/article/details/76684698

Leetcode 专栏收录该内容

90 篇文章

订阅专栏

本文介绍了一种高效的矩阵置零算法，该算法能在O(m+n)的空间复杂度内完成任务，避免了使用额外的大规模空间。文章详细解释了如何利用矩阵的第一行和第一列来标记哪些行和列需要被置零，并提供了具体的实现代码。

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

题目：

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

Follow up:

Did you use extra space?
A straight forward solution using O(mn) space is probably a bad idea.
A simple improvement uses O(m + n) space, but still not the best solution.
Could you devise a constant space solution?

如果用一个二维数组记录0出现的位置，那么空间复杂度就为O(mn)

所以用第一行加第一列记录0的位置，并进行判断，空间复杂度为O(m+n)，时间复杂度为O(mn）

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int row=matrix.size();
        int col=matrix[0].size();
        bool row_flag=false,col_flag=false;
        for(int i=0;i<col;i++){
            if(matrix[0][i]==0){
                row_flag=true;
                break;
            }
        }
        for(int i=0;i<row;i++){
            if(matrix[i][0]==0){
                col_flag=true;
                break;
            }
        }
        for(int i=1;i<row;i++){
            for(int j=1;j<col;j++){
                if(matrix[i][j]==0){
                    matrix[0][j]=0;
                    matrix[i][0]=0;
                }
            }
        }
        for(int i=1;i<row;i++){
            for(int j=1;j<col;j++){
                if(matrix[i][0]==0||matrix[0][j]==0)
                    matrix[i][j]=0;
            }
        }
        if(row_flag){
            for(int i=0;i<col;i++){
                matrix[0][i]=0;
            }
        }
        if(col_flag){
            for(int i=0;i<row;i++){
                matrix[i][0]=0;
            }
        }
    }
};