Set Matrix Zeroes

最新推荐文章于 2020-02-09 12:21:36 发布

原创最新推荐文章于 2020-02-09 12:21:36 发布 · 255 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#array #java

leetcode 专栏收录该内容

175 篇文章

订阅专栏

本文介绍了一种高效的矩阵置零算法，当矩阵中的某个元素为0时，将该元素所在的整行和整列都置为0。算法通过巧妙利用第一行和第一列作为标记位来减少额外的空间消耗。

摘要生成于 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.

解题思路：遍历(m-1)*(n-1)个元素，若matrix[i][j]=0,则使matrix[0][j]=0,matrix[i][0]=0。然后遍历第0行的n-1个元素，若matrix[0][j]=0，则j所在的一列全置为0；遍历第0列的m-1个元素，若matrix[i][0]=0,则i所在的一行全置为0。
关键之所在就是这里的第0行，第0列怎么处理，必须的先记录下第0行或第0列刚开始是否有0，只有刚开始有0存在的时候，最后才可以把第0行或第0列的数置为0。

代码如下：

public void setZeroes(int[][] matrix) {
    int m=matrix.length,n=matrix[0].length;
    boolean row0=false,col0=false;
    if(matrix[0][0]==0){
        row0=true;  col0=true;
    }else{
        for(int j=1;j<n;j++){
            if(matrix[0][j]==0){
                row0=true;break;
            }
        }
        for(int i=1;i<m;i++){
            if(matrix[i][0]==0){
                col0=true;break;
            }
        }
    }
    for(int i=1;i<m;i++){
        for(int j=1;j<n;j++){
            if(matrix[i][j]==0){
                matrix[i][0]=0;
                matrix[0][j]=0;
            }
        }
    }
    for(int j=1;j<n;j++){
        if(matrix[0][j]==0){
            for(int i=1;i<m;i++){
                matrix[i][j]=0;
            }
        }
    }
    for(int i=1;i<m;i++){
        if(matrix[i][0]==0){
            for(int j=1;j<n;j++){
                matrix[i][j]=0;
            }
        }
    }
    if(row0){
        for(int j=0;j<n;j++){
            matrix[0][j]=0;
        }
    }
    if(col0){
        for(int i=0;i<m;i++){
            matrix[i][0]=0;
        }
    }
}