数字序列中某一位的数字

题目描述

来源于 https://leetcode-cn.com/

数字以0123456789101112131415…的格式序列化到一个字符序列中。在这个序列中,第5位(从下标0开始计数)是5,第13位是1,第19位是4,等等。

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

示例 1:

输入:n = 3
输出:3

示例 2:

输入:n = 11
输出:0

限制:

  • 0 <= n < 2^31

注意:本题与主站 400 题相同:https://leetcode-cn.com/problems/nth-digit/

解法:

class Solution {
public:
    int findNthDigit(int n) {
        if(n < 10) return n;

        int width = 0;
        int count = 9;
        int i = 1;
        int len = 0;
        while(i + len <= n){
            i += len;
            width += 1;
            
            // 直接执行 len = count * width 会在最后一次
            // 循环的时候发生溢出,因此在这里先做判断
            if(count > (n - i) / width){
                break;
            }
            
            len = count * width;
            count *= 10;

        }
        int num = pow(10, width-1) + (n - i) / width;
        int m = (n - i) % width;
        for(int k=width-1;k>m;k--){
            num = num / 10;
        }
        return num % 10;
    }
};