有序子集测试

msa*_*aio 5 python tuples subset python-itertools

我想测试一个有序集是否是一个更大的有序集的子集.我使用了元组和itertools.combinations:

def subset_test(a, b):
    return a in itertools.combinations(b, len(a))
Run Code Online (Sandbox Code Playgroud)

例如,

>>> subset_test((0, 1, 2), (0, 3, 1, 4, 2))
True
>>> subset_test((0, 1, 2), (0, 3, 2, 4, 1))
False
Run Code Online (Sandbox Code Playgroud)

它有效,但是当我测试大元组时它很慢.

Joh*_*ooy 12

您可以简单地使用迭代器来跟踪B中的位置

>>> A = (0, 1, 2)
>>> B = (0, 3, 1, 4, 2)
>>> b_iter = iter(B)
>>> all(a in b_iter for a in A)
True
Run Code Online (Sandbox Code Playgroud)