51Nod 1042 數字0-9的數量(前導0的數位dp)

題目鏈接：https://vjudge.net/contest/374535#problem/F
給出一段區間a-b，統計這個區間內0-9出現的次數。
比如 10-19，1出現11次（10,11,12,13,14,15,16,17,18,19,其中11包括2個1)，其餘數字各出現1次。

Input
兩個數a,b（1 <= a <= b <= 10^18)

Output
輸出共10行，分別是0-9出現的次數

Sample Input

10 19

Sample Output

1
11
1
1
1
1
1
1
1
1

完整代碼：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
int a[30];
LL dp[30][30][30];
LL dfs(int pos,int sta,int lead,int limt,int number)///位數 狀態  前導0 最高位限制 目標爲number
{
    if(pos==-1)
        return sta;///所有位已經搜索完，返回找到的number的數量
    if(!lead&&!limt&&dp[pos][number][sta]!=-1)
        return dp[pos][number][sta];
    int up=limt?a[pos]:9;
    LL res=0;
    for(int i=0; i<=up; i++)
    {
        if(lead&&i==0)///前導0標記存在，並且當前位也爲0,當前位也當做前導0
            res+=dfs(pos-1,sta,lead,limt&&i==up,number);
        else
            res+=dfs(pos-1,sta+(i==number),lead&&i==0,limt&i==up,number);///i==number的值爲0或1
    }
    if(!limt&&!lead)
        dp[pos][number][sta]=res;
    return res;
}
LL solve(LL x,int number)
{
    int tot=0;
    while(x)
    {
        a[tot++]=x%10;
        x/=10;
    }
    return dfs(tot-1,0,1,1,number);
}
int main()
{
    LL m,n;
    scanf("%lld%lld",&m,&n);
    memset(dp,-1,sizeof(dp));
    for(int i=0; i<=9; i++)///統計區間[m,n]每一個i的數量
        printf("%lld\n",solve(n,i)-solve(m-1,i));
    return 0;
}