入门力扣自学笔记182 C++ （题目编号：902）（数位DP模板）

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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

思路：

以 n = 123456 为例，预设条件有：
1、先将 n 逐位拆分，存到 digit 数组中
2、limit 表示当前枚举到的位是否有大小限制，如果当前数字为 12345? ，则 ? 处的位limit 为 1 ，因为这里取值范围是 [1,6]；如果当前数字为 12344? ，则 limit 为 0 ，取值范围是[1,9]
3、lead 表示是否有前导零，如果当前数字为 0000?? ，则 lead 为 1
4、f [pos] [limit] [lead]用于记忆化搜索

基本方法就是枚举 + 记忆化搜索，逐步枚举 n的所有数位，根据当前状态进行递归搜索，具体实现见代码

---

代码：

class Solution {
public:
    int digit[50];
    int f[11][2][2];
    // 用来快速检索数字是否在digits中出现
    vector<bool> num;
    int dfs(int pos, int limit, int lead){
        // 搜索完毕，只要不是全是0的情况就返回1
        if(pos == -1) return !lead;
        if(f[pos][limit][lead] != -1) return f[pos][limit][lead];
        // big表示搜索上限，和 limit 及 n 中当前位数字有关
        int big = limit ? digit[pos] : 9;
        int sum=0;
        for(int i=0;i<=big;i++){
            if(!num[i]) continue;
            // 这里要注意如果当前数字有0，只能以前导0的形式出现，其他情况都不合法
            if(i==0 && !lead) continue;
            sum += dfs(pos-1, limit && i==big, lead && i==0);
        }
        f[pos][limit][lead] = sum;
        return sum;
    }
    int atMostNGivenDigitSet(vector<string>& digits, int n) {
        // 将 n 逐位拆分，存到 digit 数组中
        int p=0;
        while(n){
            digit[p++] = n % 10;
            n /= 10;
        }
        num.resize(10,false);
        num[0] = true;
        for(string& s : digits){
            num[s[0]-'0'] = true;
        }
        memset(f,-1,sizeof(f));
        return dfs(p-1, 1, 1);
    }
};

---

作者：good-thing
链接：https://leetcode.cn/problems/numbers-at-most-n-given-digit-set/solution/jing-dian-by-good-thing-fwau/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。