Leetcode 233. Number of Digit One

题目描述：给定n,找出小于n的数字中含1的个数。

题目链接：Leetcode 233. Number of Digit One

reference: https://leetcode.com/discuss/44281/4-lines-o-log-n-c-java-python

intuitive: 每10个数, 有一个个位是1, 每100个数, 有10个十位是1, 每1000个数, 有100个百位是1. 做一个循环, 每次计算单个位上1得总个数(个位,十位, 百位).

以算百位上1为例子: 假设百位上是0, 1, 和 >=2 三种情况:

• case 1: n=3141092, a= 31410, b=92. 计算百位上1的个数应该为 3141 *100 次.
• case 2: n=3141192, a= 31411, b=92. 计算百位上1的个数应该为 3141 *100 + (92+1) 次.
• case 3: n=3141592, a= 31415, b=92. 计算百位上1的个数应该为 (3141+1) *100 次.

以上三种情况可以用 一个公式概括:

(a + 8) / 10 * m + (a % 10 == 1) * (b + 1);

代码如下

class Solution:
    def countDigitOne(self, n):
        """
        :type n: int
        :rtype: int
        """
        ans = 0
        m = 1
        while(m<=n):
            a = n//m
            b = n%m
            ans += (a+8)//10 * m
            if(a%10 == 1): ans += (b+1)
            m *= 10
        return ans

参考链接

• leetcode 233: Number of Digit One