角矩形的数量

/*
 * 思路：
 * 1. 使用Java Stream API简化代码。
 * 2. 遍历每一对行，计算每一列中同时为1的个数。
 * 3. 使用组合数学计算每一对行之间形成的矩形的数量。
 */
import java.util.stream.IntStream;

public class CornerRectangles {
    public int countCornerRectangles(int[][] grid) {
        int m = grid.length;
        int n = grid[0].length;
        return IntStream.range(0, m)
                .mapToObj(i -> i)
                .flatMap(i -> IntStream.range(i + 1, m)
                        .mapToObj(j -> new int[]{i, j}))
                .mapToInt(pair -> {
                    int i = pair[0];
                    int j = pair[1];
                    long pairCount = IntStream.range(0, n)
                            .filter(k -> grid[i][k] == 1 && grid[j][k] == 1)
                            .count();
                    return (int) (pairCount * (pairCount - 1) / 2);
                })
                .sum();
    }

    public static void main(String[] args) {
        CornerRectangles solution = new CornerRectangles();
        int[][] grid = {
            {1, 0, 0, 1},
            {0, 0, 1, 0},
            {1, 0, 0, 1},
            {1, 1, 1, 1}
        };
        System.out.println(solution.countCornerRectangles(grid)); // 输出: 1
    }
}