【数位DP】

2.SGU 258

problem

一个2*n位数，前n位数各位数和与后n位数各位数和相等，是lucky数。一个2*n位数，改变一个数字后，依然是2*n位(改前改后都没有前导零)，并且是lucky数，则改之前的数称之为近似lucky数。求[l, r] 区间内，有多少近似lucky数。

think

dp[枚举到那一位][这个数是2*n位数][sum(前n位各位数和-后n位数各位数和)][more(最多增加)][less(最多减少)]

sum可能<0 所以给他都+45

more = max(9-前n位某个数, 后n位某个数)

less = max(首位-1, 前n位且非首位的某个数, 9-后n位某个数) //因为首位不能变成0 所以首位-1

答案是sum!=45(是0 的话就不用变了) && sum+more >=45 && sum-less<=45 

code

[cpp]  view plain copy print ?

1. const int M = 45;  
2. int f[10][10][91][10][10];  
3. int b[11];  
4.   
5. int dfs(int pos, int N, int sum, int more, int less, int e){  
6.     if(pos==0){  
7.         return sum!=M && sum+more>=M && sum-less<=M;  
8.     }  
9.     if(!e && ~f[pos][N][sum][more][less]) return f[pos][N][sum][more][less];  
10.     int ans = 0;  
11.     int u = e?b[pos]:9;  
12.     int d = pos==N?1:0;  
13.     for(; d<=u; d++){  
14.         int ss = pos>N/2 ? sum+d : sum-d;  
15.         int mm = pos>N/2 ? max(more, 9-d) : max(more, d);  
16.         int ll = pos>N/2 ? max(less, pos==N?d-1:d) : max(less, 9-d);  
17.         ans += dfs(pos-1, N, ss, mm, ll, e&&d==u);  
18.     }  
19.     return e ? ans : f[pos][N][sum][more][less] = ans;  
20. }  
21.   
22. int s(int n){  
23.     if(n==-1) return 0;  
24.     int p = 0;  
25.     while(n){  
26.         b[++p] = n%10;  
27.         n/=10;  
28.     }  
29.     int ans = 0;  
30.     for(int i=2; i<=p; i+=2){  
31.         ans += dfs(i, i, 0+M, 0, 0, i==p);  
32.     }  
33.     return ans;  
34. }  
35.   
36. int main(){  
37.     memset(f, -1, sizeof(f));  
38.     int a, b;  
39.     while(~scanf("%d%d", &a, &b)){  
40.         printf("%d\n", s(b)-s(a-1));  
41.     }  
42.     return 0;  
43. }  

3.URAL 1057

problem

[l, r] 区间内，有多少个数分解成K个不同B的次方。

think

平时写数位DP，习惯把数位按十进制分解，这道题，分解成B进制，就可以了。然后看有多少个数里面有K个1 其他都是0.

code

[cpp]  view plain copy print ?

1. int K, B;  
2. int f[50][50];  
3. int bit[50];  
4.   
5. int dfs(int pos, int num, bool e){  
6.     if(pos==0) return num==K;  
7.     if(!e && ~f[pos][num]) return f[pos][num];  
8.     int ans = 0;  
9.     int u = e?bit[pos]:B-1;  
10.     for(int d=0; d<=1 && d<=u; d++){  
11.         ans += dfs(pos-1, num+d, e&&d==u);  
12.     }  
13.     return e?ans:f[pos][num]=ans;  
14. }  
15.   
16. int s(int n){  
17.     int p = 0;  
18.     while(n){  
19.         bit[++p] = n%B;  
20.         n/=B;  
21.     }  
22.     return dfs(p, 0, true);  
23. }  
24.   
25. int main(){  
26.     int n, m;  
27.     while(~scanf("%d%d%d%d", &n, &m, &K, &B)){  
28.         memset(f, -1, sizeof(f));  
29.         printf("%d\n", s(m)-s(n-1));  
30.     }  
31.     return 0;  
32. }