leetcode-400. 第N个数字

题目

在无限的整数序列 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, …中找到第 n 个数字。

注意:
n 是正数且在32为整形范围内 ( $n<2^{31}$)。

示例 1:

输入:
3

输出:
3

示例 2:

输入:
11

输出:
0

说明:
第11个数字在序列 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, … 里是0，它是10的一部分。

解题思路

这个题……真是一言难尽，头条和作业帮都出过，我一直没想清楚，现在重新做一下。

其实就是判断n是落在哪个区间的数学题，关键在于要明确好各个区间所占的位数。
对于1位数而言，共有1～9，9 * 1=9个字符
对于2位数而言，有10～99，共90 * 2个字符
对于3位数，有900 * 3个字符

所以先判断n落在几位数的区间，再判断n比初始值高多少，算最终指向哪个数字。

成绩

时间>96.03%
空间>52.04%

代码

class Solution:
    def findNthDigit(self, n: int) -> int:
        upper_bound = 9
        digit_num = 1
        # upper_bound是如果n能到达digit_num位数，那么至少要有多大。如n>9 + 90 * 2, 则说明n至少是在3位数里
        while n > upper_bound:
            upper_bound += (10 ** digit_num * 9 * (digit_num + 1))
            digit_num += 1
        # lower_bound是本digit_num中的下界, 用来计算n在该digit_num区间中的第几个数字
        lower_bound = upper_bound - (10 ** (digit_num - 1) * 9 * digit_num) + 1
        above = n - lower_bound
        
        base_num = int(('9' * (digit_num - 1)) if digit_num > 1 else '0') + 1
        number = base_num + above // digit_num
        index = above % digit_num
        return int(str(number)[index])