leetcode73.矩阵置零

方法一:采用两个数组,行数组记录哪些行的所有元素需要全部置为0,列数组需要记录哪些列的所有元素需要全部置为0

java 复制代码
public class Solution {
    public void setZeroes(int[][] matrix) {
        boolean[] rowArr = new boolean[matrix.length];
        boolean[] colArr = new boolean[matrix[0].length];
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[0].length; j++) {
                if(matrix[i][j]==0){
                    rowArr[i]=true;
                    colArr[j]=true;
                }
            }
        }
        for (int i = 0; i < rowArr.length; i++)
            if(rowArr[i])
                for(int j=0;j<matrix[0].length;j++)
                    matrix[i][j]=0;
        for(int j=0;j<colArr.length;j++)
            if(colArr[j])
                for (int i = 0; i < matrix.length; i++)
                    matrix[i][j]=0;
    }
}

方法二:在方法一的基础上优化空间复杂度,用第一行和第一列代替方法一中的行数组和列数组,但是这样会导致第一行变动,结果错误,为此引入两个变量,一个记录第一行是否需要全部置为0,另一个记录第一列是否需要全部置为0

java 复制代码
public class Solution {
    public void setZeroes(int[][] matrix) {
        int m= matrix.length;
        int n=matrix[0].length;
        boolean rowZero=false,colZero=false;
        for (int[] ints : matrix) {
            if (ints[0] == 0) {
                colZero = true;
                break;
            }
        }
        for (int i = 0; i < n; i++) {
            if(matrix[0][i]==0){
                rowZero=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 i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if(matrix[i][0]==0||matrix[0][j]==0)
                    matrix[i][j]=0;
            }
        }
        if(rowZero)
            for (int i = 0; i < n; i++) {
                matrix[0][i]=0;
            }
        if(colZero)
            for (int i = 0; i < m; i++) {
                matrix[i][0]=0;
            }
    }
}
相关推荐
Phoebe鑫15 分钟前
数据结构每日一题day7(顺序表)★★★★★
算法
ffyyhh9955111 小时前
idea自动生成注释
java·ide·intellij-idea
showpingzhang1 小时前
Intellij IDEA2023 创建java web项目
java·intellij-idea
酷爱码1 小时前
python和Java的区别
java·开发语言
Cloud_.2 小时前
Spring Boot整合Redis
java·spring boot·redis·后端·缓存
盖世英雄酱581362 小时前
Java开发工程师必须掌握的线程知识指南
java
Suc_zhan2 小时前
实验二 如何将随机森林算法应用于激酶抑制剂分类任务
python·算法·随机森林·机器学习
小学仔2 小时前
leetcode -编辑距离
java·算法·动态规划
AI是这个时代的魔法2 小时前
Using Dyck Path to solve a leetcode puzzle
python·算法·leetcode
愚戏师2 小时前
数据结构与算法分析:树与哈希表(一)
数据结构·算法·链表·深度优先·广度优先·宽度优先