统计元音字母序列的数目

/*
 * 思路：
 * 使用流的方式处理相同的动态规划问题。
 * 虽然 Java Stream 并不完全适合复杂的动态规划问题，
 * 但我们可以使用它来简化一些操作，如初始化和求和。
 */
import java.util.stream.LongStream;

public class VowelPermutationStream {
    private static final int MOD = 1000000007;

    public int countVowelPermutation(int n) {
        long[] dpA = new long[n];
        long[] dpE = new long[n];
        long[] dpI = new long[n];
        long[] dpO = new long[n];
        long[] dpU = new long[n];

        // 初始化第一行
        dpA[0] = dpE[0] = dpI[0] = dpO[0] = dpU[0] = 1;

        LongStream.range(1, n).forEach(i -> {
            dpA[(int)i] = (dpE[(int)i-1] + dpI[(int)i-1] + dpU[(int)i-1]) % MOD;
            dpE[(int)i] = (dpA[(int)i-1] + dpI[(int)i-1]) % MOD;
            dpI[(int)i] = (dpE[(int)i-1] + dpO[(int)i-1]) % MOD;
            dpO[(int)i] = dpI[(int)i-1] % MOD;
            dpU[(int)i] = (dpI[(int)i-1] + dpO[(int)i-1]) % MOD;
        });

        long result = (dpA[n-1] + dpE[n-1] + dpI[n-1] + dpO[n-1] + dpU[n-1]) % MOD;
        return (int) result;
    }
}