安排电影院座位

/*
 * 思路：
 * 1. 使用 Java Stream API 处理 reservedSeats 数组，将其转为一个 Map。
 * 2. 对于每一行，检查是否可以安排 4 人家庭在三个可能的位置：
 *    - 座位 2 到 5
 *    - 座位 4 到 7
 *    - 座位 6 到 9
 * 3. 根据每一行可以安排的 4 人家庭数量进行累加。
 */

import java.util.*;
import java.util.stream.*;

public class CinemaSeatAllocationStream {
    public int maxNumberOfFamilies(int n, int[][] reservedSeats) {
        Map<Integer, Set<Integer>> reservedMap = Arrays.stream(reservedSeats)
            .collect(Collectors.groupingBy(seat -> seat[0], Collectors.mapping(seat -> seat[1], Collectors.toSet())));

        int maxFamilies = reservedMap.values().stream().mapToInt(reserved -> {
            boolean canPlaceLeft = !reserved.contains(2) && !reserved.contains(3) && !reserved.contains(4) && !reserved.contains(5);
            boolean canPlaceMiddle = !reserved.contains(4) && !reserved.contains(5) && !reserved.contains(6) && !reserved.contains(7);
            boolean canPlaceRight = !reserved.contains(6) && !reserved.contains(7) && !reserved.contains(8) && !reserved.contains(9);
            int count = 0;
            if (canPlaceLeft) count++;
            if (canPlaceRight) count++;
            else if (canPlaceMiddle) count++;
            return count;
        }).sum();

        return maxFamilies + 2 * (n - reservedMap.size());
    }
}