1~n 整数中 1 出现的次数

题目:

输入一个整数 n, 求 1 ~ n 这 n 个整数的十进制表示中 1 出现的次数.

问题分析:

当 $n$ 是最大的 $k$ 位数时, 即 $n = maxNum^k$

$$n = 10^{k+1} - 1, sum(n) = k \cdot 10^{k-1}$$

任何一位上的数字都可以是 $0\sim 9$ (位数不足的在前面补 $0$, 如四位数的$1$表示为$0001$ ), 可知, $0 \sim n$ 共有 $k\cdot 10^k$ 个 $0\sim 9$ 之间的数字, 每个数字出现的次数必然相等, 所以有上式成立. 可以用以下方式表示, 令

$$h(k) = 10\cdot h(k-1), h(1) = 1$$

则

$$h(k) = 10^{k -1}, k > 0$$

故有

$$sum(maxNum^k) = k\cdot h(k), k \ge 1$$

为了讨论的方便, 将 $n$ 记为 $(n_kn_{k-1}...n_1)$, 其中 $0 \le n_i \le 9, n_k\neq 0, i = k, k-1, ..., 1$. 对于任意的 $n_i$ 有如下公式:

$$sum(n_i) = n_i\cdot (i-1)\cdot h(i-1)+ \begin{cases} 0, & \text{$n_i=0$} \\[2ex] n \mod h(i) + 1, & \text{$n_i = 1$} \\[2ex] h(i), & \text{$n_i>1$} \end{cases}$$

则可得

$$sum(n) = \sum_{i=1}^k sum(n_i)$$

案例:

$n = 10531$

计算过程
个位数 1 : $sum(1_1) = 1*(1-1)*h(1-1) + n\mod h(1)_{1_1=1} + 1 = 0 + 0 + 1 = 1$ ;
十位数 3 : $sum(3_2) = 3 * (2 - 1) * h(2-1) + h(2)_{3_2>1} = 3 + 10 = 13$;
百位数 5 : $sum(5_3) = 5 * (3 - 1) * h(3-1) + h(3)_{5_3>1} = 100 + 100 = 200$;
千位数 0 : $sum(0_4) = 0 * (4 - 1) * h(4-1) + 0_{0_4=0} = 0 + 0 = 0$
万位数 1 : $sum(1_5) = 1 * (5 - 1) * h(5-1) + n\mod h(5)_{1_5=1} + 1 = 4000 + 531 + 1 = 4532$

则 $sum(n)$ 如下:

$$sum(n) = \sum_{i=1}^5 sum(n_i)=1+13 +200+0+4532=4746$$

根据上述思想及公式, C++代码实现如下:

long NumberOf1(const int n)
{
    long sum = 0;
    int m = n;

    for (int k = 0, hk = 1; m > 0; ++k)
    {
        int modK = m % 10;
        sum = modK * k * (hk / 10);

        if (modK == 1)
            sum += n % hk + 1;
        else if (modK > 1)
            sum += hk;

        m /= 10;
        hk *= 10;
    }

    return sum;
}

该算法的时间复杂度为 O(logN), 空间复杂度为 O(1).