P2602 [ZJOI2010]数字计数--数位dp,dfs

题目

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 105;
const ll MOD = 1e9 + 7;
const ll INF = 0x7fffffff;
ll dp[N][N];
ll nums[N];//保存数位
ll dfs(ll len, ll pre, ll x, bool ismax, bool iszero) {//len为长度，pre记录和，ismax表示上一位数是否取到了为最高位
    if (!len)return pre;//递归边界，长度为0时返回pre
    if (dp[len][pre] != -1 && !ismax && !iszero)return dp[len][pre];//如果找过了,且不是最高位,且无前导零直接返回结果
    ll top = ismax ? nums[len] : 9;//当前最高位是多少，例如n=655，假如最高位是6，第二位最大只能是5=num[len]，否则最大可以是 9。
    ll ans = 0;
    for (ll i = 0; i <= top; i++) {
        if (iszero && i == 0)ans += dfs(len - 1, pre, x, ismax && i == top, iszero);
        else ans += dfs(len - 1, pre + (i == x), x, ismax && i == top, false);
    }
    if (!ismax && !iszero)dp[len][pre] = ans;//记忆化
    return ans;
}
ll solve(ll n,ll num) {
    ll len = 0;
    while (n)nums[++len] = n % 10, n /= 10;//取出每一位
    memset(dp, -1, sizeof(dp));//初始化-1（因为有可能某些情况下的方案数是0）
    return dfs(len, 0, num, true, true);
}
int main() {
    ios_base::sync_with_stdio(false), cin.tie(0), cout.tie(0);

    ll l, r;
    cin >> l >> r;
    for (ll i = 0; i <= 9; i++)
    cout << solve(r,i) - solve(l - 1,i) << " ";

    return 0;
}