题目描述
给定一个整数n,计算所有小于等于n的非负整数中数字1出现的个数。
示例1
输入:n = 13
输出:6
示例2
输入:n = 0
输出:0
提示
- 0 <= n <= 2*109
思路:递归
递归
依据题意,需要统计[0, n]范围内所有整数中,数字1出现的个数。
先对数字观察规律如下:
9以下的1有1个。
99以下的1有20个,其中既包含了尾数为1的9个两位数,又包括以1开头的10个两位数,还有1这个数,总共有20个1。
999以下的有100 + 10 * 20 =300。
以此类推。给定一个数字n,首先要明确它能出现多少个小于等于它自身的999…。以2234为例,其中完整包含0~999的有999、1999,一共两次,则需要计算两次999以下的1的个数,并加上1000-1999中千位数字1的个数。最后还有2000-2234的部分共有235个数中的1的个数。
class Solution:
def countDigitOne(self, n: int) -> int:
# 递归
if n < 10:
return 1 if n else 0
num = str(n)
x = len(num) - 1
nxt = int(num[1:])
res = self.compute(x) * int(num[0]) + self.countDigitOne(nxt)
return res + 10 ** x if int(num[0]) > 1 else res + nxt + 1
@lru_cache(None)
def compute(self, x):
return 10 ** (x-1) + 10 * self.compute(x-1) if x else 0