LeetCode-零矩阵

最新推荐文章于 2024-07-20 22:34:59 发布

氤氲国度

最新推荐文章于 2024-07-20 22:34:59 发布

阅读量52

点赞数

分类专栏： LeetCode 文章标签： leetcode 矩阵算法

本文链接：https://blog.csdn.net/weixin_43607515/article/details/134229698

版权

LeetCode 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

#LeetCode

题目描述

![[Pasted image 20231103163422.png]]

初次解法

暴力解法
- 使用标记数组

class Solution {

    public void setZeroes(int[][] matrix) {

        // 暴力解法

        // 找出需要置零的行数与列数

        int[] colFlagArr = new int[matrix[0].length];

        int[] rowFlagArr = new int[matrix.length];

        for(int i = 0; i < matrix.length; i++){

            for(int j = 0; j < matrix[0].length; j++){

                if(matrix[i][j]==0){

                    colFlagArr[j] = 1;

                    rowFlagArr[i] = 1;

                }

            }  

        }

        // 行置零

        for(int i = 0; i < rowFlagArr.length; i++){

            if(rowFlagArr[i] == 1){

                for(int j = 0; j < matrix[0].length; j++){

                    matrix[i][j] = 0;

                }

            }

        }

        // 列置零

        for(int i = 0; i < colFlagArr.length; i++){

            if(colFlagArr[i] == 1){

                for(int j = 0; j < matrix.length; j++){

                    matrix[j][i] = 0;

               }

            }

        }

        return;

    }

}

优化解法（官方解一）

利用矩阵的第一行和第一列来取代暴力解法中的标记数组
另外起两个布尔值来分别标记第一行和第一列是否包含零值

class Solution {

    public void setZeroes(int[][] matrix) {

        // 使用布尔值来取代标记数组

        boolean row_zero = false;

        boolean col_zero = false;

        for(int i = 0; i < matrix.length; i++){

            if(matrix[i][0] == 0) col_zero = true;

        }

  

        for(int i = 0; i < matrix[0].length; i++){

            if(matrix[0][i] == 0) row_zero = true;

        }

        for(int i = 1; i < matrix.length; i++){

            for(int j = 1; j < matrix[0].length; j++){

                if(matrix[i][j]==0){

                     matrix[i][0] = 0;

                     matrix[0][j] = 0;

                }

            }  

        }

  

        for(int i = 1; i < matrix.length; i++){

            for(int j = 1; j < matrix[0].length; j++){

                if(matrix[i][0]==0 || matrix[0][j]==0){

                     matrix[i][j] = 0;

                }

            }  

        }

        if(row_zero) {

            for(int j = 0; j < matrix[0].length; j++){

                matrix[0][j]=0;

            }

        }

        if(col_zero) {

            for(int j = 0; j < matrix.length; j++){

                matrix[j][0]=0;

            }

        }

        return;

    }

}