【剑指Offer】1～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

题解

根据当前位 curcur 值的不同，分为以下三种情况：

当 cur = 0 时： 此位 1 的出现次数只由高位 high 决定，计算公式为：high×digit

如下图所示，以 n = 2304为例，求 digit = 10(即十位)的 11 出现次数。

当 cur = 1时： 此位 11 的出现次数由高位 high 和低位 low 决定，计算公式为：high×digit+low+1

如下图所示，以 n = 2314 为例，求 digit = 10digit=10 (即十位)的 1出现次数。

当 cur = 2, 3, …, 9时： 此位 1 的出现次数只由高位 high决定，计算公式为：(high+1)×digit

如下图所示，以 n = 2324 为例，求 digit = 10(即十位)的 1 出现次数。

class Solution {
    public int countDigitOne(int n) {
        int digit=1,res=0;
        int high=n/10,low=0,cur=n%10;
        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;
    }
}