HDU 3652 B-number(数位DP)

时间:2023-03-09 01:06:35
HDU 3652 B-number(数位DP)

题目链接

学习大神的数位DP模版。

 #include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int dp[][][];
int num[];
//dp[i][j][k] 代表前i位余数为j时候 flag的3种情况
int dfs(int pos,int pre,int flag,int bound)//pos代表位置,pre代表余数,
{
int ans = ,end,tflag,tpre,i;
if(pos == -)//flag 1代表上位是1,2代表前面已经有13了,0代表其他的。
return pre == &&flag == ;
if(!bound&&dp[pos][pre][flag] != -)
return dp[pos][pre][flag];
end = bound ? num[pos] : ;
for(i = ;i <= end;i ++)
{
tpre = (pre* + i)%;
tflag = flag;
if(flag == &&i == )
tflag = ;
else if(flag == &&i == )
tflag = ;
else if(flag == &&i != )
tflag = ;
ans += dfs(pos-,tpre,tflag,bound&&i == end);
}
if(!bound)
dp[pos][pre][flag] = ans;
return ans;
}
int judge(int x)
{
int pos = ;
while(x)
{
num[pos++] = x%;
x = x/;
}
return dfs(pos-,,,);
}
int main()
{
int n;
memset(dp,-,sizeof(dp));
while(scanf("%d",&n)!=EOF)
printf("%d\n",judge(n));
return ;
}