51Nod 1042 数字0-9的数量 (数位DP

1042 数字0-9的数量

基准时间限制：1 秒 空间限制：131072 KB 分值: 10 难度：2级算法题
给出一段区间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出现的次数
Input示例
10 19
Output示例
1
11
1
1
1
1
1
1
1
1

0这个数字需要单独考虑下 要不然算不出来啊啊

#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
#define ll long long
const int mod = 1e9+7;
ll dp[100];
ll Pow (ll a, ll b)    //手写 pow函数 防止精度缺失 
{
    ll ans = 1;
    while(b) {
        if(b&1) ans = ans*a;
        b /= 2;
        a = a*a;
    }
return ans;
}
void init()
{
    memset(dp,0,sizeof(dp));
    for(int i = 1;i <= 18; i++) {
        dp[i] = i*Pow(10,i-1);                //计算出现的次数 比如00 到 99 都是20 
    }
} 
ll sum(ll x, ll k)
{
    ll radix = 1, tail = 0, res = 0;
    ll ans = x;
    ll digit = 0, len = 0;
    while(x) {
        len++;
        digit = x%10;
        x /= 10;
        if(digit > k) {
            res += radix + digit*dp[len-1];; 
        }
        else  if(digit == k) {
            res += tail + 1 + digit*dp[len-1];
        }
        else {
            res += digit*dp[len-1];
        }
        tail += digit*radix;
        radix *= 10;
    }
    if(k == 0) {                 // 去除重复的个数  比如 000-999 每次都重复计算了0的个数 需要减掉 
        ll m = 1;
        while(ans) {
            ans /= 10;
            res -= m;
            m *= 10;
        }
    }
return res;
} 


int main()
{
        ll a, b;
        while(~scanf("%lld%lld",&a,&b)) {
            init();
            for(int i = 0;i <= 9; i++) {
                printf("%lld\n",sum(b,i)-sum(a-1,i));
            }
        }
return 0;
}