水题数位dp 0~9

题意：记录a到b 的数字，0~9的出现的次数
数位dp水题，但是自己找了很久没找到错，
最后才发现是因为没有加sta位置的状态错了。。。

LL dp[20][20][15][20];
int num[30];
LL dfs(int sta,int pos,int limit,int dig,int cnt){
    if(pos<0){return cnt;}
    if(!limit&&~dp[sta][pos][dig][cnt])return dp[sta][pos][dig][cnt];
    int up=limit?num[pos]:9;
    LL ret=0;
    for(int i=0;i<=up;++i){
        if(sta==pos&&i==0&&dig==0)ret+=dfs(sta-1,pos-1,limit&&i==up,dig,cnt);
        //else if(sta!=pos&&i==0&&dig==0)ret+=dfs(sta,pos-1,limit&&i==up,0,cnt+1);
        else ret+=dfs(sta,pos-1,limit&&i==up,dig,cnt+(i==dig?1:0));
        if(dig==0){
            for(int j=sta;j<=pos;++j)cout<<num[j];cout<<i;puts("");
        }
    }
    if(!limit)dp[sta][pos][dig][cnt]=ret;
    return ret;
}
LL ans[20];
LL solve(LL x,int dig){
    int pos=0;
    while(x){
        num[pos++]=x%10;
        x/=10;
    }
    return dfs(pos-1,pos-1,1,dig,0);
}
int main(){
    LL a,b;
    mem(dp,-1);
    while(~sf("%lld%lld",&a,&b)){
        for(int i=0;i<=9;++i)ans[i]=solve(b,i)-solve(a-1,i);
        pf("%lld %lld\n",solve(b,0),solve(a,0));
        for(int i=0;i<=9;++i)pf("%lld%c",ans[i],i==9?'\n':' ');
    }
}