Day39 矩阵置零

最新推荐文章于 2021-11-23 11:01:19 发布

微光长明

最新推荐文章于 2021-11-23 11:01:19 发布

阅读量281

点赞数 1

分类专栏： LeetCode刷题集文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/walkmanfy/article/details/114384033

版权

LeetCode刷题集专栏收录该内容

90 篇文章 0 订阅

订阅专栏

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

https://leetcode-cn.com/problems/set-matrix-zeroes/

示例1：

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

示例2：

输入:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
输出:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]

提示：

一个直接的解决方案是使用 O(mn) 的额外空间，但这并不是一个好的解决方案。
一个简单的改进方案是使用 O(m + n) 的额外空间，但这仍然不是最好的解决方案。
你能想出一个常数空间的解决方案吗？

Java解法

思路：

设想是记录为0 的坐标位置，然后按为0 位置进行列、行置零
使用set分别记录位置
但空间复杂度没有达到要求

package sj.shimmer.algorithm.m2;

import java.util.HashSet;
import java.util.Set;

import sj.shimmer.algorithm.Utils;

/**
 * Created by SJ on 2021/3/4.
 */

class D39 {
    public static void main(String[] args) {
        int[][] matrix = {{0,1,2,0}, {3,4,5,2}, {1,3,1,5}};
        setZeroes(matrix);
        Utils.logArrays(matrix);
    }
    public static void setZeroes(int[][] matrix) {
        if (matrix==null||matrix.length==0||matrix[0]==null||matrix[0].length==0) {
            return;
        }
        Set<Integer> zeroColumn = new HashSet<>();
        Set<Integer> zeroLine = new HashSet<>();
        int lineLength = matrix.length;
        int columnLength = matrix[0].length;
        for (int i = 0; i < lineLength; i++) {
            for (int j = 0; j < columnLength; j++) {
                if (matrix[i][j]==0) {
                    zeroLine.add(i);
                    zeroColumn.add(j);
                }
            }
        }
        for (int i = 0; i < lineLength; i++) {
            if (zeroLine.contains(i)) {
                matrix[i] = new int[columnLength];
                continue;
            }
            for (Integer integer : zeroColumn) {
                matrix[i][integer] = 0;
            }
        }
    }
}