矩阵中 1 的最大数量

/*
 * 题目思路:
 * 给定一个 m x n 的矩阵，其中矩阵的元素值为 0 或 1。
 * 我们需要找到矩阵中每行和每列都至少包含一个 1 的最大数量。
 * 这意味着我们要尽可能多地在矩阵中放置 1，但不能违反行和列的限制。
 */

import java.util.stream.IntStream;

public class Solution {
    public int maximumOnes(int[][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        int[] rowSum = new int[m];
        int[] colSum = new int[n];

        // 计算每行和每列中的 1 的数量
        IntStream.range(0, m).forEach(i ->
            IntStream.range(0, n).forEach(j -> {
                if (matrix[i][j] == 1) {
                    rowSum[i]++;
                    colSum[j]++;
                }
            })
        );

        // 统计行和列中 1 的最大数量
        long maxOnes = IntStream.range(0, m).flatMap(i ->
            IntStream.range(0, n).filter(j ->
                matrix[i][j] == 1 && rowSum[i] > 1 && colSum[j] > 1
            )
        ).count();

        return (int) maxOnes;
    }
}