lintcode(三)：统计数字 【中等题】python

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

样例

输入：
k = 1, n = 12
输出：
5
解释：
在 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12] 中，我们发现 1 出现了 5 次 (1, 10, 11, 12)(注意11中有两个1)。

思路1
遍历[0,n]，看每一个数字各位所对应的数和k是否相等，如果相等，count+1
优点：代码好写
缺点：时间复杂度为O(n)，LintCode用时106ms

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
        if n==0 and k==0:
            return 1
        
        count = 0
        for num in range(n+1):
            while num!=0:
                cur = num%10
                num = num//10
                if cur == k :
                    count += 1
        if k == 0 :
            count += 1
        
        return count

思路2
将此问题看成k分别在[0,n]中各位的全排列
优点：时间复杂度为O(logn)
缺点：要考虑的情况比较多，代码相对难写