力扣 233. 数字 1 的个数 数学 计数

https://leetcode-cn.com/problems/number-of-digit-one/

思路：答案就是$n$的每一位（个位、十位、百位…）上出现的1的数量的总和。那么对于任意一个数字$abcde$，假设计算$c$位上的1，且有$high=ab,cur=c,low=de,digit=10^2$，现在开始分类讨论$c$的取值：

1. $c=0$，为了让这一位取1，同时整个数字不能大于$abcde$，高位$high$只能取$[0,high-1]$，此时低位可以任意取，因此1的总数为：$high\ast digit$。举个例子，$n=1204$，为了让十位取1，高位只能取$[0,11]$，此时低位可以取$[0,9]$任意一个数，因此1的总数为：12*10=120。
2. $c=1$，情况1仍然成立，但是当高位$high$取最大时，低位不能任意取了，最多只能取到$de$，因此1的总数为：$high\ast digit+low+1$。举个例子，$n=1214$，考虑$high=12,cur=1$的情况，$low$只能取$[0,4]$，因此1的总数为：12*10+5=125。
3. $c>1$，情况1依然成立，且高位$high$取最大、该位取1时，低位依然能够任意取：$high\_1\_any<higt\_c\_low$，因此1的总数为：$(high+1)\ast digit$。

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