题解 | [ZJOI2010]COUNT 数字计数

#include <iostream>
#include <vector>
using namespace std;
typedef long long ll;

// 计算每个数字在1到n中出现的次数
vector<ll> count_digits(ll n) {
    vector<ll> cnt(10, 0);
    ll pos = 1;  // 位权：1, 10, 100, ...
    
    while (n >= pos) {
        ll high = n / (pos * 10);  // 高位数字
        ll curr = (n / pos) % 10;  // 当前位数字
        ll low = n % pos;          // 低位数字
        
        // 对0-9每个数字统计在当前位出现的次数
        for (int digit = 0; digit < 10; digit++) {
            // 计算当前位之前的完整循环次数
            cnt[digit] += high * pos;
            
            // 处理当前位
            if (digit < curr) {
                cnt[digit] += pos;
            } else if (digit == curr) {
                cnt[digit] += low + 1;
            }
            
            // 处理前导零的特殊情况
            if (digit == 0) {
                cnt[0] -= pos;  // 减去前导零的情况
            }
        }
        
        pos *= 10;
    }
    
    return cnt;
}

int main() {
    ll a, b;
    cin >> a >> b;
    
    // 计算[1,b]和[1,a-1]的统计结果
    vector<ll> cnt_b = count_digits(b);
    vector<ll> cnt_a = count_digits(a - 1);
    
    // 输出[a,b]区间的结果
    for (int i = 0; i < 10; i++) {
        cout << cnt_b[i] - cnt_a[i] << " ";
    }
    cout << endl;
    
    return 0;
}

import java.util.Scanner;

public class Main {
    // 计算每个数字在1到n中出现的次数
    static long[] countDigits(long n) {
        long[] cnt = new long[10];
        long pos = 1;  // 位权：1, 10, 100, ...
        
        while (n >= pos) {
            long high = n / (pos * 10);  // 高位数字
            long curr = (n / pos) % 10;  // 当前位数字
            long low = n % pos;          // 低位数字
            
            // 对0-9每个数字统计在当前位出现的次数
            for (int digit = 0; digit < 10; digit++) {
                // 计算当前位之前的完整循环次数
                cnt[digit] += high * pos;
                
                // 处理当前位
                if (digit < curr) {
                    cnt[digit] += pos;
                } else if (digit == curr) {
                    cnt[digit] += low + 1;
                }
                
                // 处理前导零的特殊情况
                if (digit == 0) {
                    cnt[0] -= pos;  // 减去前导零的情况
                }
            }
            
            pos *= 10;
        }
        
        return cnt;
    }
    
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        long a = sc.nextLong();
        long b = sc.nextLong();
        
        // 计算[1,b]和[1,a-1]的统计结果
        long[] cnt_b = countDigits(b);
        long[] cnt_a = countDigits(a - 1);
        
        // 输出[a,b]区间的结果
        for (int i = 0; i < 10; i++) {
            System.out.print(cnt_b[i] - cnt_a[i] + " ");
        }
        System.out.println();
        
        sc.close();
    }
}

def count_digits(n):
    cnt = [0] * 10
    pos = 1  # 位权：1, 10, 100, ...
    
    while n >= pos:
        high = n // (pos * 10)  # 高位数字
        curr = (n // pos) % 10  # 当前位数字
        low = n % pos          # 低位数字
        
        # 对0-9每个数字统计在当前位出现的次数
        for digit in range(10):
            # 计算当前位之前的完整循环次数
            cnt[digit] += high * pos
            
            # 处理当前位
            if digit < curr:
                cnt[digit] += pos
            elif digit == curr:
                cnt[digit] += low + 1
            
            # 处理前导零的特殊情况
            if digit == 0:
                cnt[0] -= pos  # 减去前导零的情况
        
        pos *= 10
    
    return cnt

# 读取输入
a, b = map(int, input().split())

# 计算[1,b]和[1,a-1]的统计结果
cnt_b = count_digits(b)
cnt_a = count_digits(a - 1)

# 输出[a,b]区间的结果
print(*[cnt_b[i] - cnt_a[i] for i in range(10)])