矩阵置零

/*
 * Problem Statement:
 * Given an m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.
 * 
 * Approach:
 * 1. Identify rows and columns that need to be zeroed using an extra marker array.
 * 2. Use Java Streams to simplify the code and make it more readable.
 */
 
import java.util.stream.IntStream;
 
public class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        boolean[] zeroRows = new boolean[m];
        boolean[] zeroCols = new boolean[n];
 
        // Find rows and columns that need to be zeroed
        IntStream.range(0, m).forEach(i -> {
            IntStream.range(0, n).forEach(j -> {
                if (matrix[i][j] == 0) {
                    zeroRows[i] = true;
                    zeroCols[j] = true;
                }
            });
        });
 
        // Zero out rows
        IntStream.range(0, m).forEach(i -> {
            if (zeroRows[i]) {
                IntStream.range(0, n).forEach(j -> matrix[i][j] = 0);
            }
        });
 
        // Zero out columns
        IntStream.range(0, n).forEach(j -> {
            if (zeroCols[j]) {
                IntStream.range(0, m).forEach(i -> matrix[i][j] = 0);
            }
        });
    }
}