HDU 3652 B-number(数位DP)

题目链接

学习大神的数位DP模版。

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 using namespace std;
 5 int dp[31][31][31];
 6 int num[31];
 7 //dp[i][j][k] 代表前i位余数为j时候 flag的3种情况
 8 int dfs(int pos,int pre,int flag,int bound)//pos代表位置，pre代表余数，
 9 {
10     int ans = 0,end,tflag,tpre,i;
11     if(pos == -1)//flag 1代表上位是1，2代表前面已经有13了，0代表其他的。
12     return pre == 0&&flag == 2;
13     if(!bound&&dp[pos][pre][flag] != -1)
14     return dp[pos][pre][flag];
15     end = bound ? num[pos] : 9;
16     for(i = 0;i <= end;i ++)
17     {
18         tpre = (pre*10 + i)%13;
19         tflag = flag;
20         if(flag == 1&&i == 3)
21         tflag = 2;
22         else if(flag == 0&&i == 1)
23         tflag = 1;
24         else if(flag == 1&&i != 1)
25         tflag = 0;
26         ans += dfs(pos-1,tpre,tflag,bound&&i == end);
27     }
28     if(!bound)
29     dp[pos][pre][flag] = ans;
30     return ans;
31 }
32 int judge(int x)
33 {
34     int pos = 0;
35     while(x)
36     {
37         num[pos++] = x%10;
38         x = x/10;
39     }
40     return dfs(pos-1,0,0,1);
41 }
42 int main()
43 {
44     int n;
45     memset(dp,-1,sizeof(dp));
46     while(scanf("%d",&n)!=EOF)
47     printf("%d\n",judge(n));
48     return 0;
49 }

 

posted @ 2013-09-18 14:30  Naix_x  阅读(244)  评论(0)  编辑  收藏  举报