acwing 338.计数问题 数位dp

给定两个整数 a 和 b，求 a 和 b 之间的所有数字中 0∼9 的出现次数。

例如，a=1024，b=1032，则 a 和b 之间共有 9 个数如下：

1024 1025 1026 1027 1028 1029 1030 1031 1032

其中 0 出现 10 次，1 出现 10 次，2 出现 7 次，3 出现 3 次等等…

输入格式

输入包含多组测试数据。

每组测试数据占一行，包含两个整数 a 和 b。

当读入一行为  0 时，表示输入终止，且该行不作处理。

输出格式

每组数据输出一个结果，每个结果占一行。

每个结果包含十个用空格隔开的数字，第一个数字表示 0 出现的次数，第二个数字表示 1 出现的次数，以此类推。

数据范围

0<a,b<100000000

输入样例：

1 10
44 497
346 542
1199 1748
1496 1403
1004 503
1714 190
1317 854
1976 494
1001 1960
0 0

#include<bits/stdc++.h>
using namespace std;
const int N=10;
//采用前缀和的思路
int get(vector<int>num,int l,int r)  //计算l位和r位的组成的数字
{
    int res=0;
    for(int i=l;i>=r;i--)
    {
        res=res*10+num[i];
    }
    return res;
}
int power10(int x) //算10的x次方
{
    int res=1;
    while(x--)
    {
        res*=10;
    }
    return res;
}
int count(int n,int x) // 算出数字从1到n，x出现的次数
{
    if(!n) return 0;//如果n为0,直接返回0即可。
    vector<int>num; //开一个数组存储每位上的数字
    while(n)
    {
        num.push_back(n%10);
        n/=10;
    }
    n=num.size(); //以上操作就是将一个整数，把各位上的数字存在一个num数组里
    int res=0;//存储答案
    for(int i=n-1-!x;i>=0;i--)   //从高位开始 枚举前面位为0到abc-1,当x取0时应该从第二高位开始，因为x=0时，最高位不可能取0
    {
       if(i<n-1) 
       {
           res+=get(num,n-1,i+1)*power10(i);   //如果i非最高位，此时res加上x所在位前面的数值abc*10的i次方
           if(!x)  res-=power10(i);   //如果x为0，则前面的数值应该是001开始到abc-1，总共abc-1个数，所以应该减去一个10的i次方
       }
       if(num[i]==x) //i为最高位的时候，也可以说前面为abc的时候
       {
           res+=get(num,i-1,0)+1; //此时efg只要比原来的小即可，即0到efg一共efg+1个数；
       }
       else if (num[i]>x)  res+=power10(i) ;  //如果该位上的数大于x，则后面所以任意选，是10的i次方个数
    }
    return res;
}
int main()
{
    int a,b;
    while(cin>>a>>b,a)
    {
        if(a>b) swap(a,b); //如果a>b，则两数交换，求从b到a的数位
        for(int i=0;i<=9;i++)
        {
            cout<<count(b,i)-count(a-1,i)<<' ';
        }
        cout<<endl;
    }
    return 0;
}