python - 在递增和并列的整数序列中找到给定位置 i 的整数

Question

披露:这是为了帮助家庭作业

我想在给定位置 i 找到一个整数，同时反复构建和添加一个整数序列，最好具有良好的运行时间和性能。

Answer 1

您在 while 的每次迭代中从 1 开始重建子序列，而不是简单地保留一个序列并在接下来的迭代中添加下一个数字，然后扩展 主列表。

此外，您应该将 str.join 推迟到 while 之后，而不是在每次迭代时构建字符串:

from itertools import count

def give_output(digitPos):
    c = count(1)
    l, lst = [], []
    while len(lst) <= digitPos:
        l.append(next(c))  # update previous sub-sequence
        lst.extend(l)
    return int(''.join(map(str, lst))[digitPos-1])

---

时间:

In [10]: %%timeit
    ...: giveOutput(500)
    ...: 
1000 loops, best of 3: 219 µs per loop

In [11]: %%timeit
    ...: give_output(500)
    ...: 
10000 loops, best of 3: 126 µs per loop

大约一半的时间!

如果您使用 div-mod 方法而不是构建一个大字符串来选择第 i 个项目，您甚至可以做得更好；我会把它留给你。