python - 获取连接范围列表的索引

Question

我有一个范围列表，该列表中的所有范围都具有相同的 start和 stop ，但不一样 step .
例如:

[range(0, 10, 2), range(0, 10, 3)]

当然，列表可以包含不止 2 个范围。
连接的范围列表代表以下数字:

[0, 2, 3, 4, 6, 8, 9]

我想得到 x连接范围列表的索引。
例如，最后一个示例的 5 索引将是 8 .

问题是范围可能很大(数百万)，我不想将该范围变成一个列表来获取 x指数。我需要以某种方式计算该值 x没有“打开”那个范围列表的索引

尝试了几个小时来想出一种算法来做到这一点，我发现的最佳解决方案包括使用二分搜索来做到这一点，我认为这不是理想的方法。

知道我怎样才能做到这一点吗？

Answer 1

您可以重新制作 range 同 start和 end ，并打包所有 step s 到一个新列表。现在在范围内的每个数字上，您可以检查它是否匹配 any步。你可以把它变成一个生成器:

def steps_range(start, end, steps):
    for i in range(start, end):
        if any(i % x == 0 for x in steps):
            yield i

现在您可以在该生成器上循环，直到到达相关索引。根据你的例子:

ranges = [range(0, 10, 2), range(0, 10, 3)]

start = ranges[0].start
end = ranges[0].stop
steps = [r.step for r in ranges]

target_index = 5

for i, num in enumerate(steps_range(start, end, steps)):
    print(num)
    if i == target_index:
        break

这将打印出:

0
2
3
4
6
8