//len是现在正在处理的位数
//state是现在的状态如果等于6则state是1,下一位受限
//fp是看上一位是否受限,如果上一位受限,则这一位也要受限
int dfs(int len,bool state,bool fp)
{
if(!len)//当len==0时,搜索结束,返回是否找到了答案
return 1;
//DP里保存的是完整的,也即不受限的答案,所以如果满足的话,可以直接返回。
if(!fp && dp[len][state] != -1)
return dp[len][state];
int ret = 0 , fpmax = fp ? digit[len] : 9;//fpmax存储当前位能够到达的最大值
for(int i=0;i<=fpmax;i++)
{
if(i == 4 || state && i == 2)//含有不吉利数62或4的直接跳过
continue;
ret += dfs(len-1,i == 6,fp && i == fpmax);
}
if(!fp)
dp[len][state] = ret;//DP里保存完整的、取到尽头的数据
return ret;
}
int f(int n)
{
int len = 0;
while(n)
{
digit[++len] = n % 10;//digit里面存储的是边界的每一位
n /= 10;
}
return dfs(len,false,true);
}
数位DP模板(改)
最新推荐文章于 2022-02-11 21:51:45 发布