学生出勤记录 II

/*
 * 思路：
 * - 使用流式处理来计算合法出勤记录的数量。
 * - 我们使用一个包含所有可能的长度为 n 的出勤记录的流。
 * - 然后过滤出满足条件的记录：
 *   - 缺勤次数小于 2
 *   - 不存在连续 3 天或以上的迟到记录
 * - 最后统计符合条件的记录数量，并取模 (10^9 + 7)。
 */
 
import java.util.stream.Stream;
 
public class AttendanceRewardStream {
    private static final int MOD = 1000000007;
 
    public int checkRecord(int n) {
        return (int) Stream.generate(() -> new char[n])
                .limit((int) Math.pow(3, n))
                .parallel()
                .map(this::generateRecord)
                .filter(this::isValid)
                .count() % MOD;
    }
 
    private String generateRecord(char[] chars) {
        int index = 0;
        for (int i = 0; i < chars.length; i++) {
            if (index % 3 == 0) chars[i] = 'P';
            else if (index % 3 == 1) chars[i] = 'L';
            else chars[i] = 'A';
            index++;
        }
        return new String(chars);
    }
 
    private boolean isValid(String record) {
        int absentCount = 0;
        int lateStreak = 0;
        for (char c : record.toCharArray()) {
            if (c == 'A') absentCount++;
            if (c == 'L') lateStreak++;
            else lateStreak = 0;
            if (absentCount >= 2 || lateStreak >= 3) return false;
        }
        return true;
    }
}