数位DP的记忆化搜索形式比一般递推形式好写多了,而且一般不容易出错。
dfs求[0,n]有多少个符合的,先把n换成字符串形式。
cur:现在处理到哪一位。
s:搜索到目前为止,之前的状态,具体什么状态看情况而定。
e:是否到达边界,如果没到这一位只取到9,否则只能取到bit[cur]。
z:前导0标记,也就是是否前面的都是0,有些情况需要特殊判断。
一般情况下dp数组开二维,dp[cur][s]为e和z都为0的情况,因为e和z都为0的情况比较多。特殊情况开三维什么的都是可能的。
当cur<0时判断当前状态是否符合,符合返回1,不符合返回0。
每次根据s和当前的取值i计算出新的s。
int dfs(int cur,int s,int e,int z){
if(cur<0) return check(s);
if(!e&&!z&&dp[cur][s]!=-1) return dp[cur][s];
int end=e?bit[cur]:9;
int ans=0;
for(int i=0;i<=end;i++){
if(z&&!i) ans+=dfs(cur-1,0,e&&i==end,1);
else ans+=dfs(cur-1,get_news(s,i),e&&i==end,0);
}
if(!e&&!z) dp[cur][s]=ans;
return ans;
}
int solve(int n){
int len=0;
while(n){
bit[len++]=n%10;
n/=10;
}
return dfs(len-1,0,1,1);
}