0数位DP困难 LeetCode902. 最大为 N 的数字组合

902. 最大为 N 的数字组合

描述

给定一个按 非递减顺序 排列的数字数组 digits 。你可以用任意次数 digits[i] 来写的数字。例如，如果 digits = [‘1’,‘3’,‘5’]，我们可以写数字，如 ‘13’, ‘551’, 和 ‘1351315’。

返回 可以生成的小于或等于给定整数 n 的正整数的个数 。

示例 1：

输入：digits = ["1","3","5","7"], n = 100
输出：20
解释：
可写出的 20 个数字是：
1, 3, 5, 7, 11, 13, 15, 17, 31, 33, 35, 37, 51, 53, 55, 57, 71, 73, 75, 77.
示例 2：

输入：digits = ["1","4","9"], n = 1000000000
输出：29523
解释：
我们可以写 3 个一位数字，9 个两位数字，27 个三位数字，
81 个四位数字，243 个五位数字，729 个六位数字，
2187 个七位数字，6561 个八位数字和 19683 个九位数字。
总共，可以使用D中的数字写出 29523 个整数。
示例 3:

输入：digits = ["7"], n = 8
输出：1
 

提示：

1 <= digits.length <= 9
digits[i].length == 1
digits[i] 是从 '1' 到 '9' 的数
digits 中的所有值都 不同 
digits 按 非递减顺序 排列
1 <= n <= 109

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/numbers-at-most-n-given-digit-set
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

数位DP
dp[i]表示在不是边界情况下(即i到最后一位的数字可以任意选)，第i位到最后一位总共有多少种满足条件的数字个数。
本题拼凑出的整数长度若小于n的位数一定是不在边界状态的，各个数可以任意选。
只有凑出整数的长度等于n的位数时，才是边界状态。
遍历n的长度，计算每一位为起始数字的个数。
dfs的作用时计算出以idx位置为开始，到最后一个位置结束，能拼凑出多少个数字小于等于n。

class Solution {
    public int atMostNGivenDigitSet(String[] digits, int n) {
        String s = n + "";
        int[] num = new int [s.length()];
        for (int i = 0; i < num.length; i++) {
            num[i] = s.charAt(i)-'0';
        }
        int[] dp = new int[num.length];
        int ans = 0;
        for (int i = num.length - 1; i > 0; i--) {
            ans += dfs(digits,num,dp,i,false);
        }
        return ans + dfs(digits,num,dp,0,true);
    }

    public int dfs(String[] digits, int[] num, int[] dp, int idx, boolean limit) {
        if (idx == num.length) {
            return 1;
        }
        if (!limit && dp[idx] != 0) {
            return dp[idx];
        }
        int sum = 0;
        int up = limit ? num[idx] : 9;
        for (int i = 0; i < digits.length && Integer.parseInt(digits[i]) <= up; i++) {
            sum += dfs(digits,num,dp,idx+1,limit && Integer.parseInt(digits[i]) == num[idx]);
        }
        dp[idx] = sum;
        return sum;
    }
}