这个专题学习的时间跨度有些大,因为被中间的训练给隔开了,然后我发现我好像突然对树形DP有些陌生了,我是废物 ,改天再加深一下印象。
所谓“数位DP”,是指对数字的“位”进行与计算有关的DP。一个数有个位、百位、百位、千位等。数的每一位就是数位。数位DP用来解决于数字操作有关的问题,这些问题的特征是给定的区间特别大,不能用暴力的方法逐个检查,必须用接近
O
(
log
2
n
)
O(\log_2 n)
O(log2n)复杂度的算法。解题的思路是用DP对“数位”进行操作,记录算过的区间的状态,然后再后续计算中,快速进行大范围的筛选。
经典例题 不要62
一个数字,如果包含‘4’或者‘62’,他是不吉利的。给定m和n,0<m≤n<106,统计[m,n]范围内吉利的个数。
定义状态dp[i][j]
表示i位数中首位是j,符合要求的数的个数。推公式为:
d
p
[
i
]
[
j
]
=
∑
k
=
0
9
d
p
[
i
−
1
]
[
k
]
,
(
j
≠
4
)
&
&
(
k
≠
2
&
&
j
≠
6
)
dp[i][j] = \sum_{k=0}^{9}{dp[i-1][k]} , (j\neq4) \&\&(k\neq2\&\&j\neq6)
dp[i][j]=k=0∑9dp[i−1][k],(j=4)&&(k=2&&j=6)
一般递推
#include<bits/stdc++.h>
using namespace std;
int n, m, d[10];
int dp[10][10];
void init(){
dp[0][0] = 1;
for(int i = 1; i < 10; i++)
for(int j = 0; j < 10; j++)
for(int k = 0; k < 10; k ++)
if(j != 4 && (j != 6 || k != 2)) //排除4和62
dp[i][j] += dp[i-1][k];
}
int solve(int len){ //计算[0,n]区间内满足条件的数字个数
int ans = 0;
for(int i = len; i >= 1; i--){ //从高位到低位处理
for(int j = 0; j < d[i]; j++)
if(j != 4 && (j != 2 || d[i + 1] != 6))
ans += dp[i][j];
if(d[i] == 4 || (d[i + 1] == 6 && d[i] == 2)){ ans--;break;} //注意ans--。因为dp[0][0] = 1
}
return ans;
}
int getans(int x){ //记录n或m每一位上的数字
int len = 1; d[1] = x % 10;
while(x /= 10) d[++len] = x % 10;
return solve(len);
}
int main(){
init();
while(~scanf("%d%d", &n, &m) && (n || m))
printf("%d\n", getans(m) - getans(n - 1));
return 0;
}
记忆化搜索
#include<bits/stdc++.h>
using namespace std;
int a, b, num[11], dp[11][2];
int dfs(int len, int ismax, int issix){
int ans = 0;
if(!len) return 1;
if(!ismax && dp[len][issix]) return dp[len][issix];
int maxx = (ismax? num[len]: 9);
for(int i = 0; i <= maxx; i++){
if(i == 4 || issix && i == 2) continue;
ans += dfs(len - 1, ismax && i == maxx, i == 6);
}
if(!ismax) dp[len][issix] = ans;
return ans;
}
int getans(int x){
int len = 1; num[1] = x % 10;
while(x /= 10) num[++len] = x % 10;
return dfs(len, 1, 0);
}
int main(){
while(~scanf("%d%d", &a, &b) && (a + b))
printf("%d\n", getans(b) - getans(a - 1));
return 0;
}