hdu 3555 数位dp入门

题意：给定一个long long类型能够存下的数字n，统计1~n之间含有49的数字的个数；

思路：初始版本：需要记录当前位置，前一位置放了那个数字，当前是否已经包含49，是否有上界这四个信息，也就是dfs的四个参数。dfs(pos,pre,istrue,limit);

其实这种做法有点相当于把普通的数位dp的求dp过程和最后总计的过程合二为一了。

[cpp] view plain copy

print ?

1. #include <cstdio>  
2. #include <cstring>  
3. using namespace std;  
4. int T;  
5. long long n;  
6. long long dp[25][12][2];  
7. int d[25],len;  
8. long long dfs(int pos,int pre,int flag,bool limit){  
9.     if(pos<0)  
10.         return flag;  
11.     if(!limit && dp[pos][pre][flag]!=-1)  
12.         return dp[pos][pre][flag];  
13.     long long res = 0;  
14.     int last = limit?d[pos]:9;  
15.     for(int i = 0;i<=last;i++)  
16.         res += dfs(pos-1, i, flag || (pre==4&&i==9), limit&&(i==last));  
17.     if(!limit)  
18.         dp[pos][pre][flag] = res;  
19.     return res;  
20. }  
21. long long solve(long long n){  
22.     len = -1;  
23.     while(n){  
24.         d[++len] = n%10;  
25.         n/=10;  
26.     }  
27.     return dfs(len,0,0,true);  
28. }  
29. int main(){  
30.     scanf("%d",&T);  
31.     memset(dp,-1,sizeof(dp));  
32.     while(T--){  
33.         scanf("%lld",&n);  
34.         printf("%lld\n",solve(n));  
35.     }  
36.     return 0;  
37. }  

仔细想一下dp数组为什么要有pre那一维度。试想，如果传进来的时候flag已经为1，也就是说之前已经出现了49，那么此时不管pre取什么值，相应的dp都应该是一样的，也就是说dp[x][0...9][1]都是相同的。再考虑到这一位之前还没有出现49，而且前一位不是4，实际上此时不管pre取什么值也都是相同的；唯一的例外就是之前没有出现过49，而且前一位恰为4的情况。

换句话说，上述代码的dp数组的后两维实际上只需要三个状态就能够分别。分别为到d[pos]前一位已经出现了49，用2表示；d[pos]前一位没有出现49，但是前一位为4，用1表示；d[pos]前一位没有出现49，而且前一位不为4，用0表示。

相应的状态转移方程也比较好求，见如下代码：

[cpp] view plain copy

print ?

1. #include <cstdio>  
2. #include <cstring>  
3. using namespace std;  
4. long long n,dp[25][3];  
5. int T,d[25],len;  
6. long long dfs(int pos,int pre,bool limit){  
7.     if(pos<0)  
8.         return pre==2;  
9.     if(!limit && dp[pos][pre]!=-1)  
10.         return dp[pos][pre];  
11.     long long res = 0;  
12.     int last = limit?d[pos]:9;  
13.     for(int i = 0;i<=last;i++){  
14.         if(pre == 0 || (pre==1&&i!=9))  
15.             res += dfs(pos-1,i==4,limit&&(i==last));  
16.         else  
17.             res += dfs(pos-1, 2, limit&&(i==last));  
18.     }  
19.     if(!limit)  
20.         dp[pos][pre] = res;  
21.     return res;  
22. }  
23. long long solve(long long n){  
24.     len = -1;  
25.     while(n){  
26.         d[++len] = n%10;  
27.         n/=10;  
28.     }  
29.     return dfs(len,0,true);  
30. }  
31. int main(){  
32.     scanf("%d",&T);  
33.     memset(dp,-1,sizeof(dp));  
34.     while(T--){  
35.         scanf("%lld",&n);  
36.         printf("%lld\n",solve(n));  
37.     }  
38.     return 0;  
39. }