找到所有好字符串

/*
 * 思路：
 * 1. 使用递归方法并结合Stream进行动态规划处理。
 * 2. 定义一个递归函数来处理每一个字符位置，同时使用流的特性简化代码。
 * 3. 在递归过程中更新状态，判断是否符合好字符串的定义。
 */

import java.util.stream.Stream;

public class GoodStringCounterStream {
    private static final int MOD = 1000000007;
    private String s1, s2, evil;
    private int n, m;
    private Integer[][][] memo;

    public int countGoodStrings(int n, String s1, String s2, String evil) {
        this.s1 = s1;
        this.s2 = s2;
        this.evil = evil;
        this.n = n;
        this.m = evil.length();
        this.memo = new Integer[n][s1.length() + 1][m + 1];

        return dfs(0, 0, 0);
    }

    private int dfs(int i, int j, int k) {
        if (k == m) return 0;
        if (i == n) return 1;
        if (memo[i][j][k] != null) return memo[i][j][k];

        char start = i < s1.length() ? s1.charAt(i) : 'a';
        char end = i < s2.length() ? s2.charAt(i) : 'z';

        int result = Stream.iterate(start, c -> (char) (c + 1)).limit(end - start + 1)
                .mapToInt(c -> {
                    boolean matchS1 = (j < s1.length() && c == s1.charAt(j));
                    boolean matchS2 = (j < s2.length() && c == s2.charAt(j));
                    int newJ = j + (matchS1 ? 1 : 0) - (matchS2 ? 1 : 0);
                    int newK = (evil.charAt(k) == c) ? k + 1 : 0;
                    return dfs(i + 1, newJ, newK);
                })
                .reduce(0, (a, b) -> (a + b) % MOD);

        memo[i][j][k] = result;
        return result;
    }
}