Simply the sample code for combinations().

This commit is contained in:
Raymond Hettinger 2008-02-27 01:44:34 +00:00
parent 2ab0552b5e
commit c105289ec4

View file

@ -103,15 +103,14 @@ loops that truncate the stream.
yield tuple(pool[i] for i in vec) yield tuple(pool[i] for i in vec)
while 1: while 1:
for i in reversed(range(r)): for i in reversed(range(r)):
if vec[i] == i + n-r: if vec[i] != i + n - r:
continue break
else:
return
vec[i] += 1 vec[i] += 1
for j in range(i+1, r): for j in range(i+1, r):
vec[j] = vec[j-1] + 1 vec[j] = vec[j-1] + 1
yield tuple(pool[i] for i in vec) yield tuple(pool[i] for i in vec)
break
else:
return
.. versionadded:: 2.6 .. versionadded:: 2.6