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 longconst 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;}