2021-5-13 剑指 Offer 44. 数字序列中某一位的数字（数学规律）

注：

题目：
数字以0123456789101112131415…的格式序列化到一个字符序列中。在这个序列中，第5位（从下标0开始计数）是5，第13位是1，第19位是4，等等。

请写一个函数，求任意第n位对应的数字。

示例 1：
输入：n = 3
输出：3

示例 2：
输入：n = 11
输出：0

限制：
0 <= n < 2^31

题解：
解题思路：
将 101112 ⋯ 中的每一位称为 数位 ，记为 n ；
将 10, 11, 12,⋯ 称为 数字 ，记为 num ；
数字 10 是一个两位数，称此数字的 位数 为 2 ，记为 digit；
每 digit位数的起始数字（即：1, 10, 100,⋯），记为 start 。

观察上表，可推出各 digit 下的数位数量 count 的计算公式：
count=9×start×digit

根据以上分析，可将求解分为三步：

1. 确定 n 所在 数字 的 位数 ，记为 digit ；
2. 确定 n 所在的 数字 ，记为 num ；
3. 确定 n 是 num中的哪一数位，并返回结果。

1.确定所求数位的所在数字的位数
如下图所示，循环执行 n 减去 一位数、两位数、… 的数位数量 count ，直至 ≤count 时跳出。

由于 n 已经减去了一位数、两位数、…、(digit-1)位数的 数位数量 count ，因而此时的 n 是从起始数字 start 开始计数的。

digit, start, count = 1, 1, 9
while n > count:
    n -= count
    start *= 10 # 1, 10, 100, ...
    digit += 1  # 1,  2,  3, ...
    count = 9 * start * digit # 9, 180, 2700, ...

结论： 所求数位 ① 在某个 digit 位数中； ② 为从数字 start 开始的第 n 个数位。

2. 确定所求数位所在的数字
如下图所示，所求数位 在从数字 startstart 开始的第 [(n - 1) / digit]个 数字 中（ start 为第 0 个数字）。

num = start + (n - 1) / digit

结论： 所求数位在数字 num 中。

3. 确定所求数位在 num 的哪一数位
如下图所示，所求数位为数字 num 的第 (n - 1) % digit 位（ 数字的首个数位为第 0 位）。

s = str(num) # 转化为 string
res = int(s[(n - 1) % digit]) # 获得 num 的 第 (n - 1) % digit 个数位，并转化为 int

结论： 所求数位是 res 。

复杂度分析：
时间复杂度 O(logn) ： 所求数位 n 对应数字 num 的位数 digit 最大为 O(logn) ；第一步最多循环 O(logn) 次；第三步中将num 转化为字符串使用 O(logn) 时间；因此总体为 O(logn) 。
空间复杂度 O(logn) ： 将数字 num 转化为字符串 str(num) ，占用 O(logn) 的额外空间。

class Solution {
public:
    int findNthDigit(int n) {
        //int类型会超出范围，因此采用long类型
        long start=1;
        long digit=1;
        long count=9;
        if(n==0){
            return 0;
        }
        while(n>count){
            n-=count;
            start*=10;
            digit+=1;
            count=9*start*digit;
        }
        int num=start+(n-1)/digit;
        string str_num=std::to_string(num);
        int result=str_num[(n-1)%digit]-'0';
        return result;
    }
};