leetcode 73: Set Matrix Zeroes

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

原创最新推荐文章于 2024-08-16 09:55:14 发布 · 171 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

201 篇文章

订阅专栏

本文介绍了一种方法，通过遍历矩阵的第一行和第一列来标记需要置零的元素，从而实现矩阵中所有零元素的快速定位与置零操作，提升算法效率。

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

Save the information to the first row and first column

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m=matrix.size();
        int n=matrix[0].size();
        bool row1=0,col1=0;
        for(int i=0;i<n;i++)
            if(matrix[0][i]==0)
            {
                row1=1;
                break;
            }
        for(int i=0;i<m;i++)
            if(matrix[i][0]==0)
            {
                col1=1;
                break;
            }
        for(int i=1;i<m;i++)
            for(int j=1;j<n;j++)
            {
                if(matrix[i][j]==0)
                {
                    matrix[0][j]=0;
                    matrix[i][0]=0;
                }
            }
        for(int i=1;i<m;i++)
            for(int j=1;j<n;j++)
                if(matrix[i][j]&&(!matrix[i][0]||!matrix[0][j]))
                    matrix[i][j]=0;
        if(row1)
            for(int i=0;i<n;i++)
                matrix[0][i]=0;
        if(col1)
            for(int i=0;i<m;i++)
                matrix[i][0]=0;
    }
};