题意:
给定两个正整数a和b,求在[a,b]中的所有整数中,每个数码(digit)各出现了多少次。
做法:
显然是数位DP,我来写个dfs题解吧
首先区间[a,b]内的数量可转化为[1,b]-[1,a-1]。考虑求一个数码出现的次数,比如1出现的次数。我们首先想爆搜怎么写,然后加一个记忆化即可。
对于一个长度为len的数,从高位到低位枚举它每一位上的数字,然后计算1出现的次数。哪些东西要记到状态里去?第一,当前的位置len一定要记的。第二,你要记录当前数位有没有比num[len]小,是一个bool值,因为这个用来确定你枚举下一位的范围。第三,要记录当前1已经出现的次数。第四,由于前导0不能算,还需记录之前是否是前导0,也是一个bool值。
那么dfs就好写了,其它几个数字都一样,具体的注释见代码吧。
代码:
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int N = 15;
ll f[N][2][N][2];
int num[N]; //num来存这个数每个位子上的数码
/*
记忆化搜索。
len是当前为从高到低第几位。issmall表示当前位是否和num[len]相等,0是相等,1是不相等。
sum表示当前数字出现的次数。zero表示之前是否是前导0。d是当前在算的数码。
*/
ll dfs(int len, bool issmall, int sum, bool zero, int d)
{
ll ret = 0;
if (len == 0) return sum; //边界条件
if (f[len][issmall][sum][zero] != -1) return f[len][issmall][sum][zero]; //记忆化
for (int i = 0; i < 10; i ++){
if (!issmall && i > num[len]) break;
/*
由于我们是从高位到低位枚举的,所以如果之前一位的数码和最大数的数码相同,这一位就只能枚举到num[len];
否则如果之前一位比最大数的数码小,那这一位就可以从0~9枚举了。
*/
ret += dfs(len-1, issmall || (i<num[len]), sum+((!zero || i) && (i==d)), zero && (i == 0), d);
/*
继续搜索,数位减一,issmall的更新要看之前有没有相等,且这一位有没有相等;
sum的更新要看之前是否为前导0或者这一位不是0;
zero的更新就看之前是否为前导0且这一位继续为0;
d继续传进去。
*/
}
f[len][issmall][sum][zero] = ret;
//记忆化,把搜到的都记下来
return ret;
}
ll solve(ll x, int d)
{
int len = 0;
while (x){
num[++ len] = x%10;
x /= 10;
} //数字转数位
memset(f, -1, sizeof f); //初始化
return dfs(len, 0, 0, 1, d); //开始在第len位上,最高位只能枚举到num[len]所以issmall是0,sum=0,有前导0。
}
int main()
{
ll a, b; //注意都要开long long
scanf("%lld%lld", &a, &b);
for (int i = 0; i < 10; i ++)
printf("%lld%c", solve(b, i)-solve(a-1, i), i == 9 ? '\n' : ' ');
return 0;
}