题目:
给定一个按 非递减顺序 排列的数字数组 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)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。