73. Set Matrix Zeroes

最新推荐文章于 2020-09-28 12:41:28 发布

原创最新推荐文章于 2020-09-28 12:41:28 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

leetcode-array 专栏收录该内容

53 篇文章

订阅专栏

本文介绍了一种高效的算法，用于在矩阵中找到值为0的元素，并将这些元素所在的行和列全部置零。该方法利用了第一行和第一列作为标记来避免额外的空间开销。

方法一

将第一行和第一列作为标记。matrix[i][j]==0,将matrix[i][0]和matrix[0][j]设为0，所以matrix[0][0]要单独考虑，也要最后遍历

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int n=matrix.size();
        if(n==0)
            return ;
        
        int m=matrix[0].size();
        int i,j;
        int row=1,col=1;
        for(i=0;i<n;i++)
        {
            for(j=0;j<m;j++)
            {
                if(matrix[i][j]==0)
                {
                    matrix[i][0]=0;
                    if(i==0)
                        row=0;
                    matrix[0][j]=0;
                    if(j==0)
                        col=0;
                }
            }
        }

        for(i=1;i<n;i++)
        {
            if(matrix[i][0]==0)
            {
                for(j=1;j<m;j++)
                    matrix[i][j]=0;
            }
        }
        for(j=1;j<m;j++)
        {
            if(matrix[0][j]==0)
            {
                for(i=1;i<n;i++)
                    matrix[i][j]=0;
            }
        }
        if(row==0)
        {
            for(i=1;i<m;i++)
                matrix[0][i]=0;
        }
        if(col==0)
        {
            for(i=1;i<n;i++)
                matrix[i][0]=0;
        }
    }
};
/*
[[0]]
[[1,2,3,4],[5,0,5,3],[3,0,2,0],[2,3,4,7]]
[[0,0,0,5],[4,3,1,4],[0,1,1,4],[1,2,1,3],[0,0,1,1]]

*/