ZJOI2010统计前n个数中各个数码出现的次数

题目链接：ZJOI2010 count 数字计数

题意就是统计某个区间之内的各个数码出现的次数。明显的数位dp。
这里的做法事实上可以进一步的通过空间换时间，比如在dp记录中增加num和lead状态，这样可以减少后续的递归次数。大概可以从18ms减少到11ms。

#include <iostream>
#include <limits.h>
#include <vector>
#include <string>
#include <stdlib.h>
#include <algorithm>
#include <map>
#include <cstdlib>
#include <stack>
#include <set>
#include <cmath>
#include <queue>
#include <stdio.h>
#include <cstring>
//#include<bits/stdc++.h>

using namespace std;

typedef long long ll;

const int maxn = 20;
const int maxm = 102;
const int inf = 0x3f3f3f3f;
const ll llinf = 0x3f3f3f3f3f3f3f3f;
const double dinf = 1061109567.0;
const int mod = 1e9 + 7;

ll dp[maxn][maxn];
int a[maxn];

ll sum1[maxn], sum2[maxn];

ll dfs(int pos, ll sum, int num, bool lead, bool limit) {
    if (pos == -1) {//在前缀合法的情况下返回贡献值
        return (ll)sum;
    }
    if (!lead && !limit && dp[pos][sum] != -1)//在不保存lead的结果的前提下必须在没有前导零的情况下计算贡献，
    //否则只能在dp中设置新的状态来保存lead
        return dp[pos][sum];//dp状态的设置之所以不能仅仅使用pos和num是因为子问题不能由num来表示，
        //而只能使用sum来表示，即pos和sum能唯一标识一个子问题。当然可以增加状态来减少递归深度，
        //比如增加lead栏可以使得总时间减少大概8ms

    int up = limit ? a[pos] : 9;

    ll res = 0;
    int tmp = 0;
    for (int i = 0; i <= up; ++i) {
        if (i == 0 && lead)//如果有前导零，则不算进总贡献
            res += dfs(pos - 1, sum, num, lead, limit && (i == up));
        else//否则加入结果中
            res += dfs(pos - 1, sum + (i == num), num, lead && (i == 0), limit && (i == up));
    }
    if (!lead && !limit)
        dp[pos][sum] = res;
    return res;
}

void solve(ll x, ll sum[]) {
    int len = 0;
    while (x) {
        a[len++] = x % 10;
        x /= 10;
    }

    for (int i = 0; i < 10; i++) {
        sum[i] = dfs(len - 1, 0, i, 1, 1);
    }
}

int main() {
    // freopen("input.txt","r",stdin);
    ll n, m;
    while (scanf("%lld%lld", &m, &n) == 2 && (n != 0 && m != 0)) {
        memset(dp, -1, sizeof(dp));
        solve(n, sum1);
        solve(m - 1, sum2);
        for (int i = 0; i < 10; ++i) {
            printf("%lld", sum1[i] - sum2[i]);
            if (i < 9)
                printf(" ");
        }
        puts("");
    }
    return 0;
}