給定一個 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) 的額外空間，但這仍然不是最好的解決方案。
你能想出一個常數(shù)空間的解決方案嗎？

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;
            }
        }
    }
}

image

官方解

https://leetcode-cn.com/problems/set-matrix-zeroes/solution/ju-zhen-zhi-ling-by-leetcode/

1. 額外存儲空間方法

   類似于我的解法

   • 時間復雜度：O(M×N)

• 空間復雜度：O(M+N)

2. O(1)空間的暴力

   某數(shù)據(jù)為0時，將該行列不為0的數(shù)據(jù)設置為虛擬值（約束范圍之外的值），遍歷完成之后，將虛擬值置為0

   這樣實現(xiàn)空間復雜度為O(1)