[Leetcode]Set Matrix Zeroes

最新推荐文章于 2023-02-25 23:03:36 发布

zhixin_wen95

最新推荐文章于 2023-02-25 23:03:36 发布

阅读量314

点赞数

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

本文链接：https://blog.csdn.net/zhixin_wen95/article/details/48993953

版权

Leetcode 专栏收录该内容

70 篇文章 0 订阅

订阅专栏

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

第0行储存有0的column，如果row有0，在检查完row后将整行设为0

public class Solution {
    public void setZeroes(int[][] matrix) {
        int nrows = matrix.length;
        int ncolumns = matrix[0].length;
        boolean firstRowHasZero = false;
        for(int i = 0; i < ncolumns; i++){
            if(matrix[0][i] == 0){
                firstRowHasZero = true;
                break;
            }
        }
        
        for(int i = 1; i< nrows; i++){
            boolean rowHasZero = false;
            for(int j = 0; j < ncolumns; j++){
                if(matrix[i][j] == 0){
                    matrix[0][j] = 0;
                    rowHasZero = true;
                }
            }
            
            if(rowHasZero){
                for(int j = 0; j < ncolumns; j++){
                    matrix[i][j] = 0;
                }
            }
        }
        
        for(int i = 0; i < ncolumns; i++){
            if(matrix[0][i] == 0){
                for(int j = 1; j < nrows; j++){
                    matrix[j][i] = 0;
                }
            }
        }
        
        if(firstRowHasZero){
            for(int j = 0; j < ncolumns; j++){
                matrix[0][j] = 0;
            }
        }
        
    }
}