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;
            }
    }
}
相关推荐
egoist20239 分钟前
【C++指南】哈希驱动的封装:如何让unordered_map/set飞得更快更稳?【上】
数据结构·c++·算法·容器·哈希算法·散列表·c++11
王有品15 分钟前
Spring MVC 核心注解与文件上传教程
java·spring·mvc
Wang2012201315 分钟前
随机深林算法是分类还是回归?
算法·分类·回归
小薛博客16 分钟前
3、整合前端基础交互页面
java·前端·ai·交互
@蓝莓果粒茶20 分钟前
LeetCode第158题_用Read4读取N个字符 II
前端·c++·python·算法·leetcode·职场和发展·c#
Heisenberg~31 分钟前
C++回溯算法详解
开发语言·c++·算法
刚入坑的新人编程32 分钟前
数据结构——栈和队列
c语言·数据结构·c++·链表·数组
RainbowSea33 分钟前
通用型产品发布解决方案(基于分布式微服务技术栈:SpringBoot+SpringCloud+Spring CloudAlibaba+Vue+ElementUI
java·spring boot·后端
YKPG40 分钟前
c语言修炼秘籍 - - 禁(进)忌(阶)秘(技)术(巧)【第六式】文件操作
java·c语言·数据库
骑驴看星星a1 小时前
P10416 [蓝桥杯 2023 国 A] XYZ
算法·职场和发展·蓝桥杯