AcWing 338. 计数问题题解

338. 计数问题 - AcWing题库

给定两个整数 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 次等等…

输入格式

输入包含多组测试数据。

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

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

输出格式

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

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

数据范围

0<a,b<100000000

分析

(如果用暴力方法做，时间复杂度为8 * 10⁸)

对于一个数 n = abcdefg，分别求1在每一位上出现的次数

例如：求1在第四位上出现的次数

1 <= xxx1yyy <= abcdefg

分为以下几种情况：

（1）xxx ∈ [000,abc - 1]，yyy ∈ [000,999]，则共有abc * 1000种情况

（2）xxx == abc

​ ①d < 1，此时abc1yyy > abc0efg，有0种情况

​ ②d = 1，yyy ∈ [000,efg]，则有efg + 1种情况

​ ③d > 1，yyy ∈ [000,999]，则有1000种情况

时间复杂度约为：10 * 2 * 8 * 10 ≈ 1600

注意：当求0在某一位上的出现次数时，需要注意两点：

• 0不能出现在第一位上
• 0前面的数字不能全部为0，例如：1<= xxx0yyy <= abcdefg，求在1到abcdefg之间0在第四位上的出现次数时，xxx取值要在[001,abc]之间
• 当xxx == abc，d = 0时，有efg + 1种情况；
• 当xxx == abc，d >= 1时，有1000种情况

而当求m在某一位上的出现次数时(m != 0),例如：1<= xxxmyyy <= abcdefg，求在1到abcdefg之间0在第四位上的出现次数时：

• 当xxx == abc，d == m时，有efg + 1种情况；
• 当xxx == abc，d > m时，有1000种情况

#include <bits/stdc++.h>

using namespace std;
int num[10];

int power10(int k)
{
    int res = 1;
    while(k){
        res *= 10;
        k --;
    }
    return res;
}

int getnum(int num[],int x,int y)
{
    int res = 0;
    for(int i = x;i <= y;i ++){
        res = res * 10 + num[i];
    }
    return res;
}

int solve(int n,int x)//求1~n中x的出现次数
{
    if(n == 0) return 0;
    int tmp = n,p = 0,res = 0,t;
    while(tmp){//获取数字n的每一位
        p ++;
        tmp /= 10;
    }
    tmp = n,t = p;
    while(tmp){//获取数字n的每一位
        num[t --] = tmp % 10;
        tmp /= 10;
    }
    for(int i = 1;i <= p;i ++){//求x在第i位中的出现次数
        if(x == 0 && i != 1){//0不可能在第一位出现
            res = res + (getnum(num,1,i - 1) - 1) * power10(p - i);
            if(num[i] < 1) res = res + getnum(num,i + 1,p) + 1;
            else res += power10(p - i); 
        }
        else if(x != 0){
            res = res + getnum(num,1,i - 1) * power10(p - i);
            if(num[i] == x) res += getnum(num,i + 1,p) + 1;
            else if(num[i] > x) res += power10(p - i); 

        }
    }
    return res;
}

int main()
{
    int a,b;
    while(cin >> a >> b,a || b){
        if(a > b) swap(a,b);
        int ans1[10] = {0},ans2[10] = {0};
        for(int i = 0;i < 10;i ++){
            int t1 = solve(b,i),t2 = solve(max(a - 1,1),i);
            if(a == 1) t2 = 0;
            cout << t1 - t2 << ' ';
        }
        cout << '\n';
    }
    return 0;
}