我正在解决涉及组合的编程难题.这让我有了一个很棒的itertools.combinations
功能,我想知道它是如何工作的.文档说该算法大致相当于以下内容:
def combinations(iterable, r): # combinations('ABCD', 2) --> AB AC AD BC BD CD # combinations(range(4), 3) --> 012 013 023 123 pool = tuple(iterable) n = len(pool) if r > n: return indices = list(range(r)) yield tuple(pool[i] for i in indices) while True: for i in reversed(range(r)): if indices[i] != i + n - r: break else: return indices[i] += 1 for j in range(i+1, r): indices[j] = indices[j-1] + 1 yield tuple(pool[i] for i in indices)
我明白了:我们从最明显的组合(r
第一个连续的元素)开始.然后我们更改一个(最后一个)项目以获得每个后续组合.
我正在努力的是一个有条件的内部for
循环.
for i in reversed(range(r)): if indices[i] != i + n - r: break
这次演习非常简洁,我怀疑这是所有魔法发生的地方.请给我一个提示,这样我就可以搞清楚.