力扣73矩阵置零

发布于:2025-02-10 ⋅ 阅读:(148) ⋅ 点赞:(0)

给定一个 m x n 的矩阵,如果一个元素为 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法

输入:matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出:[[1,0,1],[0,0,0],[1,0,1]]

输入: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的元素。

然后再处理除了第一行第一列的其他元素,进行遍历,如果有为0的元素,将第一行第一列对应的位置分别设置为0。然后再遍历第一行第一列将元素为0对应的行列(非第一行第一列)设置为0。

最后单独处理第一行第一列,根据标志位判断是否应该全置0

class Solution {
    public void setZeroes(int[][] matrix) {
        boolean firstRowZero = false;
        boolean firstColZero = false;
        //先遍历第一行
        for (int i = 0; i < matrix[0].length; i++) {
            if (matrix[0][i] == 0) {
                firstRowZero = true;
                break;
            }
        }
        //遍历第一列
        for (int i = 0; i < matrix.length; i++) {
            if (matrix[i][0] == 0) {
                firstColZero = true;
                break;
            }
        }
        //处理剩余
        for(int i = 1; i< matrix.length; i++){
            for(int j = 1; j < matrix[0].length; j++){
                if(matrix[i][j] == 0){
                    //设置对应的第一行第一列对应的位置值为0
                    matrix[0][j] = 0;
                    matrix[i][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;
                }
            }
        }
        //处理第一列
        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;
                }
            }
        }
        //处理
        if(firstRowZero){
            Arrays.fill(matrix[0], 0);
        }
        if(firstColZero){
            for(int i = 0; i < matrix.length; i++){
                matrix[i][0] = 0;
            }
        }

    }
}


网站公告

今日签到

点亮在社区的每一天
去签到