N 天后的牢房

/*
 * 思路：
 * 1. 使用Java Stream和Lambda表达式来实现相同的逻辑。
 * 2. 将每一天的状态用List和Stream的方式来处理。
 * 3. 同样的，使用Set来检测循环，避免不必要的计算。
 */

import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

public class PrisonCellsStream {
    public List<Integer> prisonAfterNDays(List<Integer> cells, int N) {
        Set<List<Integer>> seen = new HashSet<>();
        boolean hasCycle = false;
        int cycle = 0;

        for (int i = 0; i < N; i++) {
            List<Integer> next = nextDay(cells);
            if (!seen.contains(next)) {
                seen.add(next);
                cycle++;
            } else {
                hasCycle = true;
                break;
            }
            cells = next;
        }

        if (hasCycle) {
            N %= cycle;
            for (int i = 0; i < N; i++) {
                cells = nextDay(cells);
            }
        }

        return cells;
    }

    private List<Integer> nextDay(List<Integer> cells) {
        return Arrays.asList(0, 0, 1, 1, 0, 0, 0, 0).stream()
            .map(i -> (i > 0 && i < cells.size() - 1 && cells.get(i - 1).equals(cells.get(i + 1))) ? 1 : 0)
            .collect(Collectors.toList());
    }
}