hdu 3555

hdu 3555  输出含有49的数字的个数

 

数位DP

//dp[i][j]:长度为i的数的第j种状态
//dp[i][0]:长度为i但是不包含49的方案数
//dp[i][1]:长度为i且不含49但是以9开头的数字的方案数
//dp[i][2]:长度为i且包含49的方案数

(转)状态转移如下
dp[i][0] = dp[i-1][0] * 10 - dp[i-1][1];  // not include 49  如果不含49且，在前面可以填上0-9 但是要减去dp[i-1][1] 因为4会和9构成49
dp[i][1] = dp[i-1][0];  // not include 49 but starts with 9  这个直接在不含49的数上填个9就行了
dp[i][2] = dp[i-1][2] * 10 + dp[i-1][1]; // include 49  已经含有49的数可以填0-9，或者9开头的填4

接着就是从高位开始统计

在统计到某一位的时候，加上 dp[i-1][2] * digit[i] 是显然对的，因为这一位可以填 0 - (digit[i]-1)
若这一位之前挨着49，那么加上 dp[i-1][0] * digit[i] 也是显然对的。

若这一位之前没有挨着49，但是digit[i]比4大，那么当这一位填4的时候，就得加上dp[i-1][1]

 

1. #include <cstdio>  
2. #include <cstring>  
3. #include <iostream>  
4. #include <algorithm>  
5. using namespace std;  
6. typedef __int64 LL;  
7. LL dp[27][3];  
8. int c[27];  
9. //dp[i][j]:长度为i的数的第j种状态  
10. //dp[i][0]:长度为i但是不包含49的方案数  
11. //dp[i][1]:长度为i且不含49但是以9开头的数字的方案数  
12. //dp[i][2]:长度为i且包含49的方案数  
13. void init()  
14. {  
15.     memset(dp,0,sizeof(dp));  
16.     dp[0][0] = 1;  
17.     for(int i = 1; i <= 20; i++)  
18.     {  
19.         dp[i][0] = dp[i-1][0]*10-dp[i-1][1];  
20.         dp[i][1] = dp[i-1][0]*1;  
21.         dp[i][2] = dp[i-1][2]*10+dp[i-1][1];  
22.     }  
23. }  
24.   
25. int cal(LL n)  
26. {  
27.     int k = 0;  
28.     memset(c,0,sizeof(c));  
29.     while(n)  
30.     {  
31.         c[++k] = n%10;  
32.         n/=10;  
33.     }  
34.     c[k+1] = 0;  
35.     return k;  
36. }  
37. void solve(int len, LL n)  
38. {  
39.     int flag = 0;//标记是否出现过49  
40.     LL ans = 0;  
41.     for(int i = len; i >= 1; i--)  
42.     {  
43.         ans+=c[i]*dp[i-1][2];  
44.         if(flag)  
45.         {  
46.             ans+=c[i]*dp[i-1][0];  
47.         }  
48.         else if(c[i] > 4)  
49.         {  
50.             //这一位前面没有挨着49，但c[i]比4大，那么当这一位填4的时候，要加上dp[i-1][1]  
51.             ans+=dp[i-1][1];  
52.         }  
53.         if(c[i+1]==4 && c[i]==9)  
54.         {  
55.             flag = 1;  
56.         }  
57.     }  
58.     printf("%I64d\n",ans);  
59. }  
60. int main()  
61. {  
62.     int t;  
63.     LL n;  
64.     init();  
65.     scanf("%d",&t);  
66.     while(t--)  
67.     {  
68.         scanf("%I64d",&n);  
69.         int len = cal(n+1);  
70.         solve(len, n);  
71.     }  
72.     return 0;  
73. }