BZOJ 1833: [ZJOI2010]count 数字计数(在[a,b]中的所有整数中，每个数码(digit)各出现了多少次)

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

---

思路:
一维数位dp,需保留之后的数的总方案数,以及每个digit的,注意前导0的特判

---

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
int dig[18],flag[18];
struct node{
    long long Count[10],num;//num表示方案数
    void init(){
        memset(Count,0,sizeof(Count));
        num=0;
    }
}dp[18];//到了第几位,0~9分别出现的次数

void add(node &ans,node tmp){
    for(int i=0;i<10;i++)
        ans.Count[i]+=tmp.Count[i];
    ans.num+=tmp.num;
}

void print(node tmp){
    for(int i=0;i<10;i++)
        printf("**%lld ",tmp.Count[i]);
}

node dfs(int pos,int limit,int zero){
    if(pos==-1){
        node tmp;
        tmp.init();
        if(zero==0) tmp.num=1;
        return tmp;
    }
    if(limit==0&&flag[pos]==1&&zero==0)   return dp[pos];
    int last=(limit==1 ? dig[pos] : 9);
    node ans;
    ans.init();
    for(int i=last;i>=0;i--){
        if(i==0){
            node tmp=dfs(pos-1,limit&(i==last),zero&(i==0));
            if(zero==0) //前面不全为0
                ans.Count[i]+=tmp.num;
            add(ans,tmp);
        }
        else{
            node tmp=dfs(pos-1,limit&(i==last),zero&(i==0));
            ans.Count[i]+=tmp.num;
            add(ans,tmp);
        }
    }
    if(!limit&&!zero)  flag[pos]=1,dp[pos]=ans;
    return ans;
}

node solve(long long num){
    int len=0;
    while(num){
        dig[len++]=num%10;
        num/=10;
    }
    return dfs(len-1,1,1);
}

int main(){
    long long L,R;
    scanf("%lld%lld",&L,&R);
    node tmp1=solve(R),tmp2=solve(L-1);
    for(int i=0;i<10;i++)
        printf("%lld%c",tmp1.Count[i]-tmp2.Count[i],i==9? '\n':' ');
    return 0;
}