Leetcode Hot 100刷题记录 -Day14(矩阵置0)

矩阵置0

问题描述:

给定一个 mxn 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0。

示例 1:

复制代码
输入:matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出:[[1,0,1],[0,0,0],[1,0,1]]

示例 2:

复制代码
输入:matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出:[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

思路分析:

先第一次扫描数组,找到为0的元素,然后将其所在的行和列进行标记(boolean true为0);再次扫描数组,将标记为true的数组元素置为0。

java 复制代码
//提交版
class Solution {
      public int[][] setZeroes(int[][] matrix) {
        //矩阵的行数
        int m = matrix.length;
        //矩阵的列数
        int n = matrix[0].length;
        boolean[] row = new boolean[m];
        boolean[] col = new boolean[n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (matrix[i][j] == 0) {
                    row[i] = true;
                    col[j] = true;
                }
            }
        }
        for (int k = 0; k < m; k++) {
            for (int l = 0; l < n; l++) {
                if (row[k] || col[l]) {
                    matrix[k][l] = 0;
                }
            }
        }
        return matrix;
    }
}




//带有输入输出版本
import java.util.Arrays;

public class hot15_setZeroes {
    public int[][] setZeroes(int[][] matrix) {
        //矩阵的行数
        int m = matrix.length;
        //矩阵的列数
        int n = matrix[0].length;
        boolean[] row = new boolean[m];
        boolean[] col = new boolean[n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (matrix[i][j] == 0) {
                    row[i] = true;
                    col[j] = true;
                }
            }
        }
        for (int k = 0; k < m; k++) {
            for (int l = 0; l < n; l++) {
                if (row[k] || col[l]) {
                    matrix[k][l] = 0;
                }
            }
        }
        return matrix;
    }

    public static void main(String[] args){
        int[][] matrix = {{1,1,1},{1,0,1},{1,1,1}};
        System.out.println("输入:" + Arrays.deepToString(matrix));
        hot15_setZeroes hot15SetZeroes = new hot15_setZeroes();
        int[][] result = hot15SetZeroes.setZeroes(matrix);
        System.out.println("输出:" + Arrays.deepToString(result));
    }
}
相关推荐
逊嘘6 分钟前
【MySQL】表的增删改查(拓展)
java·数据库·mysql
易元16 分钟前
设计模式-代理模式
java·后端
trust Tomorrow20 分钟前
每日一题-力扣-2278. 字母在字符串中的百分比 0331
算法·leetcode
JaneYork22 分钟前
接口设计,接口返回信息,当账号不存在时候,应该返回200还是500,是直接返回R.fail()还是throw exception
java·spring boot
qianmoq27 分钟前
轻松掌握Java多线程 - 第二章:线程的生命周期
java·后端
风象南29 分钟前
Spring Boot 实现文件断点续传
java·spring boot·后端
Lecea_L34 分钟前
你能在K步内赚最多的钱吗?用Java解锁最大路径收益算法(含AI场景分析)
java·人工智能·算法
Tony8836 分钟前
热题100 - 394. 字符串解码
java·算法
cwtlw39 分钟前
JavaRedis和数据库相关面试题
java·数据库·笔记·面试
XF鸭40 分钟前
身份证正则表达式详解
java