bzoj1833: [ZJOI2010]count 数字计数

题意：

给定两个正整数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;
}