BZOJ1833或洛谷2602 [ZJOI2010]数字计数

BZOJ原题链接

洛谷原题链接

又是套记搜模板的时候。。
对\(0\sim 9\)单独统计。
定义\(f[pos][sum]\)，即枚举到第\(pos\)位，前面枚举的所有位上是当前要统计的数的个数之和为\(sum\)。

#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int N = 13;
ll f[N][N];
int a[N], nw;
inline ll re()
{
    ll x = 0;
    char c = getchar();
    bool p = 0;
    for (; c < '0' || c > '9'; c = getchar())
        p |= c == '-';
    for (; c >= '0' && c <= '9'; c = getchar())
        x = x * 10 + c - '0';
    return p ? -x : x;
}
ll dfs(int pos, int s, int lm, int zero)
{
    if (pos < 0)
        return s;
    if (!lm && !zero && f[pos][s] > -1)
        return f[pos][s];
    int i, k = lm ? a[pos] : 9;
    ll S = 0;
    for (i = 0; i <= k; i++)
        S += dfs(pos - 1, s + ((!nw && !i && !zero) || (nw && i == nw)), lm && i == a[pos], zero && !i);
    if (!lm && !zero)
        return f[pos][s] = S;
    return S;
}
ll calc(ll x)
{
    int k = 0;
    memset(f, -1, sizeof(f));
    while (x > 0)
    {
        a[k++] = x % 10;
        x /= 10;
    }
    return dfs(k - 1, 0, 1, 1);
}
int main()
{
    ll x, y;
    x = re();
    y = re();
    for (nw = 0; nw <= 9; nw++)
        printf("%lld ", calc(y) - calc(x - 1));
    return 0;
}