Python数据结构与算法（poj）：P01019——Number Sequence

OpenJudge - 01019:Number Sequence

        大意为，根据题目要求构造一个序列，并输出特定位置的数字。

        如果想直接得出整个序列，那么看一下数据范围就知道不可能，一定会超时，所以我们应采取一些数学的东西来简化，尽可能得到线性的算法。

        我们来算一下从1~k的对应长度，显然可以很容易地通过取对数再由等比数列求和并加上剩余项得到，用代码写就是这样：（其中l为总长度，所以这里写的是加上对应增量）

l += p*(10**p)-(10**p-1)//9
l += (t-10**p+1)*(p+1)

        由此得到输入索引对应1~k序列的k是多少，之后就可以打出一个数字序列（1~k），再按照输入的索引就可以输出答案。

from math import log10
def cal(x):
    t = 1
    l = 0
    while l < x:
        p = int(log10(t))
        l += p*(10**p)-(10**p-1)//9
        l += (t-10**p+1)*(p+1)
        t += 1
    tmp = ''.join(map(str, list(range(1, t))))
    return tmp[-(l-x+1)]

for _ in range(int(input())):
    print(cal(int(input())))