统计特殊子序列的数目

/*
 * 思路：
 * 使用动态规划来解决这个问题。定义三个变量 count0、count1 和 count2 分别表示以 0、1 和 2 结尾的特殊子序列的数量。
 * 遍历数组 nums，根据当前的数字更新这三个变量。
 * 1. 如果当前数字是 0，那么 count0 += 1（每个 0 都可以单独成为一个特殊序列的一部分）。
 * 2. 如果当前数字是 1，那么 count1 += count0（每个 1 可以和之前的所有 0 组成一个特殊序列）。
 * 3. 如果当前数字是 2，那么 count2 += count1（每个 2 可以和之前的所有 1 组成一个特殊序列）。
 * 最终答案就是 count2 对 10^9 + 7 取余。
 */
import java.util.stream.*;

public class Solution {
    public int countSpecialSubsequences(int[] nums) {
        int MOD = 1000000007;
        long[] counts = new long[3];
        for (int num : nums) {
            if (num == 0) {
                counts[0] = (counts[0] + 1) % MOD;
            } else if (num == 1) {
                counts[1] = (counts[1] + counts[0]) % MOD;
            } else if (num == 2) {
                counts[2] = (counts[2] + counts[1]) % MOD;
            }
        }
        return (int) (counts[2] % MOD);
    }
}