73. 矩阵置零

复制代码
https://leetcode.cn/problems/set-matrix-zeroes/description/?envType=study-plan-v2&envId=top-100-liked
复制代码
我们可以使用两个vector来记录需要置零的行和列,然后遍历这两个vector,将对应的行和列置零。这样的空间复杂度最大是O(n+m)
时间复杂度O(nm),空间复杂度O(n+m)
java 复制代码
    public void setZeroes(int[][] matrix) {
        HashSet<Integer> row = new HashSet<>();
        HashSet<Integer> col = new HashSet<>();
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[0].length; j++) {
                if (matrix[i][j] == 0) {
                    row.add(i);
                    col.add(j);
                }
            }
        }
        for(int i = 0; i < matrix.length; i++) {
            for(int j = 0; j < matrix[0].length; j++) {
                if (row.contains(i) || col.contains(j)) {
                    matrix[i][j] = 0;
                }
            }
        }
    }
复制代码
因为当我们记录i,j是需要置0的行和列时,我们可以使用原矩阵的首行和首列置0来代替集合记录,因为后续操作这些位置也要置0的.时间复杂度O(nm),空间复杂度O(1)
java 复制代码
    public void setZeroes(int[][] matrix) {
        // 定义两个布尔变量,分别表示第一列和第一行是否有0
        boolean col0 = false, row0 = false;
        // 遍历第一列,如果有0,则将col0设为true
        for (int i = 0; i < matrix.length; i++) {
            if (matrix[i][0] == 0) col0 = true;
        }
        // 遍历第一行,如果有0,则将row0设为true
        for (int i = 0; i < matrix[0].length; i++) {
            if (matrix[0][i] == 0) row0 = true;
        }
        // 遍历除第一行和第一列外的所有元素,如果有0,则将对应的第一行和第一列元素设为0
        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;
                }
            }
        }
        // 遍历第一列,如果有0,则将对应的所有行元素设为0
        for (int i = 1; i < matrix[0].length; i++) {
            if (matrix[0][i] == 0) {
                for (int j = 1; j < matrix.length; j++) {
                    matrix[j][i] = 0;
                }
            }
        }
        // 遍历第一行,如果有0,则将对应的所有列元素设为0
        for (int i = 1; i < matrix.length; i++) {
            if (matrix[i][0] == 0) {
                for (int j = 1; j < matrix[0].length; j++) {
                    matrix[i][j] = 0;
                }
            }
        }
        // 如果第一行有0,则将第一行所有元素设为0
        if(row0) {
            Arrays.fill(matrix[0], 0);
        }
        // 如果第一列有0,则将第一列所有元素设为0
        if(col0) {
            for(int i = 0; i < matrix.length; i++) {
                matrix[i][0] = 0;
            }
        }
    }
相关推荐
2401_8576363911 分钟前
Spring Boot环境下的知识分类与检索
java·spring boot·后端
阑梦清川12 分钟前
数学建模启发式算法篇(一)---遗传算法
算法·数学建模·启发式算法
小趴菜不能喝15 分钟前
spring boot 3.x 整合Swagger3
java·spring boot·swagger
ErvinHowell25 分钟前
文件MD5生成性能大提升!如何实现分片与Worker优化
前端·vue.js·算法
用户405478783748227 分钟前
深度学习笔记 - Pytorch自搭建VGG-16模型实现人脸识别
算法
微服务技术分享27 分钟前
专为成长型企业打造的Java CRM系统源码:CRM客户关系管理系统技术解析与功能构建
java·crm客户关系管理系统源码·鸿鹄crm客户关系管理系统·鸿鹄crm客户关系管理系统源码
琪露诺大湿28 分钟前
JavaEE-多线程初阶(4)
java·开发语言·jvm·java-ee·基础·1024程序员节·原神
Java程序员-小白36 分钟前
Spring Shell——快速构建终端应用,自定义终端命令
java·后端·spring
益达爱喝芬达40 分钟前
力扣11.3
算法·leetcode
y_m_h40 分钟前
leetcode912.排序数组的题解
数据结构·算法