恢复数组

/*
 * 思路：
 * 使用 Stream 和 lambda 表达式来进行动态规划计算。
 */
import java.util.stream.IntStream;

public class Solution {
    public int numberOfArrays(String s, int k) {
        int n = s.length();
        int mod = 1000000007;
        int[] dp = new int[n + 1];
        dp[0] = 1;

        IntStream.rangeClosed(1, n).forEach(i -> {
            long[] num = {0};
            IntStream.iterate(i, j -> j - 1).limit(i).forEach(j -> {
                num[0] = num[0] + (s.charAt(j - 1) - '0') * (long)Math.pow(10, i - j);
                if (num[0] > k) return;
                if (s.charAt(j - 1) != '0') {
                    dp[i] = (dp[i] + dp[j - 1]) % mod;
                }
            });
        });

        return dp[n];
    }
}