算法题笔记：统计数字 (Python3实现)

计算数字k在0到n中出现的次数，k可能是0~9之间的一个值

1.暴力法

从0到n对每个数进行逐个检测。

class Solution1:
    """
    @param k: An integer
    @param n: An integer
    @return: An integer denote the count of digit k in 1..n
    """
    def digitCounts(self, k, n):
        # write your code here
        assert(n>=0 and 0<=k<=9)
        count = 0
        for i in range(n+1):
            count+=self.singleNumCount(k,i)
        return count
    
    @staticmethod
    def singleNumCount(k,num):
        count=0
        while True:
            if num % 10 == k:
                count+=1
            num//=10
            if num == 0:
                break
        return count

2.一行法

class Solution2:
    """
    @param k: An integer
    @param n: An integer
    @return: An integer denote the count of digit k in 1..n
    """
    def digitCounts(self, k, n):
        # write your code here
        return str(list(range(n+1))).count(str(k))

3.较优解法

只通过n,分析出k出现的数量。 对于每个数，都可以从一般到特殊进行计数。例如：

1532，找4

0～1532，很明显，7在个位数上出现过153次
4>3，所以计算0～1500中7在十位数上出现的次数：150
4<5，所以计算0～1500中7在百位数上出现的次数：200
7>1，所以7没有在千位数上出现过。
所以一共出现过503次

在研究上述例子中可得如下一般结论：

每10个数，个位数上出现1次k,
每100个数，十位数上出现10次k,
每1000个数，百位数上出现100次k
…

由此结论可得基础值，再根据具体数字添加附加值

所以循环对n的右数第i位进行如下检测：
1.取第i位高一位的数字，乘以10的i-1次幂，得到基础值basic
2.取第i位数字，求附加值addition：
如果小于k，则附加值为0
如果大于k，则附加值为10的i-1次幂
如果等于k，则附加值为，右边i-1位数+1
这里可以用一个最大值最小值语句进行取值。
3.基础值+附加值为结果

另外，因为最高位不为0，所以k=0时，可以减少一次循环。除此之外，基础值需要减一个10的i-1次幂。

class Solution:
    """
    @param k: An integer
    @param n: An integer
    @return: An integer denote the count of digit k in 1..n
    """
    def digitCounts(self, k, n):
        # write your code here
        def ten(power):
            return 10**power
            
        digits = self.getNumOfDigits(n)
        
        ans = 0
        
        if k == 0:
            for i in range(digits-1):
                ans += (n // ten(i+1) -1) * ten(i)
                ans += min(n % ten(i+1), ten(i) - 1) + 1

            ans += 1
        else:
            for i in range(digits):
                ans += n // ten(i+1) * ten(i)
                ans += max(0, min( (k+1)*ten(i)-1, n%ten(i+1) ) - k*ten(i) + 1)
        
        return ans
    
    @staticmethod
    def getNumOfDigits(n):
        if n == 0:
            return 1
            
        count = 0
        while n:
            count += 1
            n //= 10
        return count

复杂度从O(n)降到O(logn)