动态规划05-数位统计类 DP

计数问题

给定两个整数$a$和$b$，求$a$和$b$之间的所有数字中$0\sim 9$的出现次数。
例如，$a=1024，b=1032$，则$a$和$b$之间共有$9$个数
如下：
$102410251026102710281029103010311032$
其中$0$出现$10$次，$1$出现$10$次，$2$出现$7$次，$3$出现$3$次等等…
输入格式
输入包含多组测试数据。
每组测试数据占一行，包含两个整数$a$和$b$。
当读入一行为$0$$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
输出样例
1 2 1 1 1 1 1 1 1 1
85 185 185 185 190 96 96 96 95 93
40 40 40 93 136 82 40 40 40 40
115 666 215 215 214 205 205 154 105 106
16 113 19 20 114 20 20 19 19 16
107 105 100 101 101 197 200 200 200 200
413 1133 503 503 503 502 502 417 402 412
196 512 186 104 87 93 97 97 142 196
398 1375 398 398 405 499 499 495 488 471
294 1256 296 296 296 296 287 286 286 247

解决思路

对于这个问题我们需要从分类讨论的角度来看，首先我们假设需要求的数字是$xxxdyyy$，下面以此举例
题目要求得到$a-b$之间的$0-9$每个数字出现的次数，我们使用$count(a,i)$函数得到从$1-a$中每一位上$i$出现的次数之和，然后利用前缀和的思想$count(b,i)-count(a-1,i)$,就得到了$a-b$之家数字i出现的次数。
接下来需要考虑的就是关于$count$函数的实现问题了，我们需要采用分类讨论的思想。
对于一个数$xxxdyyy$，假设我们当前需要求得位数是$d$，那么我们首先讨论小于前三位小于$xxx$的情况，那么我们就可以在前三位中选取$0到xxx-1$种情况，在后三位中选取$0到999$种情况，最终就是$xxx\ast 1000$种情况。
接着我们讨论当前三位等于$xxx$的情况，这种情况又可以分为第四位小于$d$，等于$d$，大于$d$。
当小于$d$时，那么就是$1000$种情况
当等于$d$时，那么就是$0$到$yyy$种情况
当大于$d$时，那就是$0$种情况
最后我们需要讨论一下关于特殊情况的问题，当我们需要求得位出现在最高位上时，那么就不需要讨论$xxx$的情况了。
当我们需要求得是数字$0$时，我们在对于$xxx$进行讨论时就不可以从$000$开始。

如果我们从$000$开始计算，那么我们就会多计算$yyy$种情况，因此减去这些情况也就是从$001$开始到$xxx-1$。

代码实现

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

//假设需要求的数字是xxx d yyy，下面以此举例

//计算num中l位到r位的数字大小
int get(vector<int> num, int l ,int r)
{
    int res = 0;
    for(int i = l; i >= r; i--)
    {
        res = res * 10 + num[i];
    }
    return res;
}

//计算10的x次方
int power10(int x)
{
    int res = 1;
    while(x--)
    {
        res *= 10;
    }
    return res;
}

int count(int n, int x)
{
    if(!n)
    {
        return 0;
    }
    vector<int> num;
    while(n)
    {
        num.push_back(n % 10);
        n /= 10;
    }
    //原来的n已经没用了，现在表示n的位数
    n = num.size();
    int res = 0;
    //放入的时候会颠倒，现在需要从最高位开始进行枚举
    //分别判断x在每一位上的数量
    // 当最高位为0时，需要从它的下一位开始读取
    for(int i = n - 1 - !x; i >= 0; i--)
    {
        //判断需要求得位数是不是最高位
        if(i < n-1)
        {
            res += get(num, n-1, i + 1) * power10(i);
            if(!x)
            {
                res -= power10(i);
            }
        }
        
        if(num[i] == x)
        {
            res += get(num,i - 1,0) + 1;
        }else if(num[i] > x)
        {
            res += power10(i);
        }
        
    }
    return res;
}

int main()
{
    int a, b;
    while(cin >> a >> b, a && b)
    {
        if(a > b)
        {
            swap(a,b);
        }
        for(int i = 0; i < 10; i++)
        {
            cout << count(b, i) - count(a - 1, i) << " ";
        }
        cout << endl;
        
    }
    return 0;
}