力扣HOT100 - 73. 矩阵置零

解题思路:

题目要求使用原地算法

在原地算法中,输入数据通常在内存中直接被修改,而不需要额外的辅助数据结构来存储中间结果。原地算法的一个常见应用是在数组或列表上进行操作,而不是创建新的数组或列表来存储结果。

java 复制代码
class Solution {
    public void setZeroes(int[][] matrix) {
        int row = matrix.length; // 矩阵的行数
        int col = matrix[0].length; // 矩阵的列数
        boolean row0_flag = false; // 标记第一行是否有零
        boolean col0_flag = false; // 标记第一列是否有零

        // 检查第一行是否有零,遍历第一行需要知道有多少列
        for (int j = 0; j < col; j++) {
            if (matrix[0][j] == 0) {
                row0_flag = true;
                break;
            }
        }

        // 检查第一列是否有零,遍历第一列需要知道有多少行
        for (int i = 0; i < row; i++) {
            if (matrix[i][0] == 0) {
                col0_flag = true;
                break;
            }
        }

        // 使用第一行和第一列作为标志位
        for (int i = 1; i < row; i++) {
            for (int j = 1; j < col; j++) {
                if (matrix[i][j] == 0) {
                    matrix[i][0] = matrix[0][j] = 0; // 如果元素为零,则将对应的第一行和第一列的元素设置为零
                }
            }
        }

        // 根据第一行和第一列的标志位,将矩阵中的元素置零
        for (int i = 1; i < row; i++) {
            for (int j = 1; j < col; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) {
                    matrix[i][j] = 0; // 如果第一行或第一列的元素为零,则将当前元素置零
                }
            }
        }

        // 如果第一行有零,则将第一行所有元素置零,遍历第一行需要知道有多少列
        if (row0_flag) {
            for (int j = 0; j < col; j++) {
                matrix[0][j] = 0;
            }
        }

        // 如果第一列有零,则将第一列所有元素置零,遍历第一列需要知道有多少行
        if (col0_flag) {
            for (int i = 0; i < row; i++) {
                matrix[i][0] = 0;
            }
        } 
    }
}
相关推荐
Chris _data2 分钟前
二叉树oj题解析
java·数据结构
牙牙7058 分钟前
Centos7安装Jenkins脚本一键部署
java·servlet·jenkins
დ旧言~14 分钟前
【高阶数据结构】图论
算法·深度优先·广度优先·宽度优先·推荐算法
paopaokaka_luck15 分钟前
[371]基于springboot的高校实习管理系统
java·spring boot·后端
张彦峰ZYF19 分钟前
投资策略规划最优决策分析
分布式·算法·金融
以后不吃煲仔饭28 分钟前
Java基础夯实——2.7 线程上下文切换
java·开发语言
进阶的架构师29 分钟前
2024年Java面试题及答案整理(1000+面试题附答案解析)
java·开发语言
The_Ticker34 分钟前
CFD平台如何接入实时行情源
java·大数据·数据库·人工智能·算法·区块链·软件工程
大数据编程之光1 小时前
Flink Standalone集群模式安装部署全攻略
java·大数据·开发语言·面试·flink
Lenyiin1 小时前
02.06、回文链表
数据结构·leetcode·链表