剑指 Offer1～n 整数中 1 出现的次数

题目描述

输入一个整数 n ，求1～n这n个整数的十进制表示中1出现的次数。
例如，输入12，1～12这些整数中包含1 的数字有1、10、11和12，1一共出现了5次。
示例 1：
输入：n = 12
输出：5
示例 2：
输入：n = 13
输出：6
限制：
1 <= n < 2^31

题目分析

我们考虑计算每一位的1的个数。即计算个十百千位。
cur是当前位，high是cur的高位，low是cur的低位，digit是当前位的因子，即10^i

令某一位为cur

cur = 0，可以理解为第cur位的高位，每一位的最低位都有一个1，但是因为cur = 0，所以cur的低位不用考虑，所以是 high×digit。

cur = 1，可以理解为第cur位的高位，每一位的最低位都有一个1，但是因为cur = 1，类似于10，12，13，但是到13就停了，因为low在限制它。所以cur的低位需要一个1，所以是 high×digit+low+1

cur = 2/3/4…/9，可以理解为第cur位的高位，每一位的最低位都有一个1，但是因为cur = 2/3/4…/9，所以cur的低位需要一个1，类似于10，11，12，…，19出现了digit个1, 所以是(high+1)×digit

注意我们循环退出的条件是 high 和 cur 都等于 0。因为都等于 0 就意味着 n 已经彻底遍历完了。不然的话：
假如退出的条件只有 high 等于 0：那假如 n = 1024，然后此时 cur 在千位，则 cur = 1，high = 0，low = 024。如果此时退出循环，我们还没来得及计算千位能出现的 1 的次数。
假如退出的条件只有 cur 等于 0：那假如 n = 1024，然后此时 cur 在百位，则 cur = 0，high = 1，low = 24。如果此时退出循环，我们还没来得及计算百位和千位能出现的 1 的次数。

代码实现

class Solution {
    public int countDigitOne(int n) {
        int digit = 1, res = 0;
        int high = n / 10, cur = n % 10, low = 0;
        while(high != 0 || cur != 0) {
            if(cur == 0) res += high * digit;
            else if(cur == 1) res += high * digit + low + 1;
            else res += (high + 1) * digit;
            low += cur * digit;
            cur = high % 10;
            high /= 10;
            digit *= 10;
        }
        return res;
    }
}