面试题43. 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

 

思路1 暴力求解

遍历1~N的整数，将其添加到一个列表中；

之后，再将列表合并为字符串，使用.count方法找到字符串中1的个数。

时间复杂度

第一步O(N)

第二步O(N)

代码

class Solution:
    def countDigitOne(self, n: int) -> int:
        ans = []
        while n:
            ans.append(str(n))
            n = n - 1
        s = "".join(ans)

        return s.count("1")

结果

正确但超时

 

思路2 找规律+递推

对任何一个整数N，都通过遍历它的每一位来得到1,2, ... , N中1出现的次数。遍历顺序为: 个位，十位，百位，千位，....

对每一位i而言，按照下述规则把N分为3个部分:

• right: 即当前位的右边。如4215，十位，则right即为42;
• left: 即当前位的左边。如4215，十位，则right即为5;
• cur: 当前位。如4215，十位，则当前位即为1.

之后，如果当前位恰好为所求出现次数的数，即恰好为1：

如4215的10位，则从1，2, ... 4215中，1在十位的情况共出现了left * 10+ (right + 1), 含义是(0,1,2, ..., 41)每个都对应了right中的(0~9)十个数；right+1表示0,1,2,3,4,5. 

如果当前位大于所求出现次数的数，即大于1：

如4215的百位，则从1，2, ... 4215中，1在十百位的情况共出现了(left + 1) * 100. 含义是(0,1,2,3,4) + 1 + (00~99)

如果当前位小于所求出现次数的数，即小于1：

则出现的次数与右边无关，而且left也只是从0~left-1， 因此出现的次数为left * base。

这样，遍历每一位即可求得结果。

 

代码

class Solution:
    def countDigitOne(self, n: int) -> int:
        if n <= 0 : return 0

        target = 1
        cnt, base = 0, 1

        while n//base != 0:
            left = n//(base * 10)
            right = n%base
            cur = (n//base)%10

            if cur == target:
                cnt += left * base + (right + 1)
            elif cur > target:
                cnt += (left + 1) * base
            else:
                cnt += left * base

            base *= 10

        return cnt

时间复杂度

O(logN)

结果