题目:2999. 统计强大整数的数目
思路:数位dp,参考大佬思路
class Solution {
public:
string low,hight;
int n;
vector<long long> v;
//参数:正在枚举的位置、受到下限的限制、受到上限的限制
long long dfs(int u,bool limit_low,bool limit_hight,string &s,int limit){
//枚举结束,符合要求
if(u==n) return 1;
//记忆化,出现过
if(!limit_low && !limit_hight &&v[u]!=-1) return v[u];
//得到当前位置值的上、下限
int left= limit_low ? low[u]-'0':0;
int right= limit_hight ? hight[u]-'0' :9;
//当前答案
long long sum=0;
//枚举的位数还没到s的长度那,前面都可以枚举
if(u<n-s.size()){
//枚举,上限也不能大于limit
for(int i=left;i<=min(right,limit);i++){
sum+=dfs(u+1,limit_low&&i==left,limit_hight && i==right,s,limit);
}
}else{
//已到s的长度范围内,必须开始匹配s了
int t=s[u-(n-s.size())]-'0';
if(left<=t&&t<=right){
sum=dfs(u+1,limit_low&&t==left,limit_hight && t==right,s,limit);
}
}
if(!limit_low && !limit_hight) v[u]=sum;
return sum;
}
long long numberOfPowerfulInt(long long start, long long finish, int limit, string s) {
//转换为字符串,好操作
low=to_string(start);
hight=to_string(finish);
n=hight.size();
//保持长度一致
low=string(n-low.size(),'0')+low;
//记忆化
v=vector<long long>(n,-1);
//参数:正在枚举的位置、受到下限的限制、受到上限的限制
return dfs(0,1,1,s,limit);
}
};