BZOJ-1833 count 数字计数 ( 数位dp )

题目：给定两个正整数a和b，求在[a,b]中的所有整数中，每个数码(digit)各出现了多少次。

思路：数位dp，sum表示digit出现的次数，num表示有多少个数。前导0需要注意下。

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
int wei[15],k;
ll a,b,ans[15],p[15];
struct node{
    ll sum,num;
};
node dp[15];

node dfs(int pos,int limit,int lead)
{
    if(pos<1)
    {
        node fuck;
        fuck.num=1;
        fuck.sum=0;
        return fuck;
    }
    if(!limit&&!lead&&dp[pos].sum!=-1)
        return dp[pos];
    int up=limit?wei[pos]:9;
    node ans,tmp;
    ans.sum=ans.num=0;
    for(int i=0;i<=up;i++)
    {
        if(i==k)
        {
            if(lead&&k==0)
            {
                tmp=dfs(pos-1,limit&&i==up,lead&&i==0);
                ans.sum+=tmp.sum;
                ans.num+=tmp.num;
                continue;
            }
            if(limit&&i==up)
            {
                tmp=dfs(pos-1,limit&&i==up,lead&&i==0);
                ans.sum+=tmp.sum+tmp.num;
                ans.num+=tmp.num;
                continue;
            }
            tmp=dfs(pos-1,limit&&i==up,lead&&i==0);
            ans.sum+=tmp.sum+tmp.num;
            ans.num+=tmp.num;
            continue;
        }
        tmp=dfs(pos-1,limit&&i==up,lead&&i==0);
        ans.num+=tmp.num;
        ans.sum+=tmp.sum;
    }
    if(!limit&&!lead)
        dp[pos]=ans;
    return ans;
}

ll solve(ll x)
{
    memset(dp,-1,sizeof dp);
    int len=0;
    while(x)
    {
        wei[++len]=x%10;
        x/=10;
    }
    node ans=dfs(len,1,1);
    return ans.sum;
}
int main()
{
    p[0]=1;
    for(int i=1;i<=13;i++)
        p[i]=p[i-1]*10;
    scanf("%lld%lld",&a,&b);
    for(k=0;k<=9;k++)
        ans[k]=solve(b)-solve(a-1);
    for(int i=0;i<9;i++)
        printf("%lld ",ans[i]);
    printf("%lld\n",ans[9]);
    return 0;
}