问题:我有一些数字列表,例如[1,1,2].我需要生成唯一的排列.排列为[1,1,2],[1,1,2],[1,2,1],[1,2,1],[2,1,1],[2,1,1] .我只需要生成唯一的排列,即[1,1,2],[1,2,1],[2,1,1].
我的尝试:我的第一个尝试是保留一组现有排列,并为itertools.permutations生成器创建一个过滤器,使用该过滤器过滤出重复项.但是,出于效率考虑,我宁愿先不要生成这些排列.即使只列出一小部分12个数字,也只有1%是唯一的.
我有一个想法,似乎无法一路搞定:我可以在列表中创建唯一值的排列,即[1,2],将其余数字放在所有不同的位置.
感谢您的任何帮助,并且要明确地说,我不想过滤掉重复的排列,我只想首先生成唯一的排列.
解决方法:
我从previous Stack Overflow answer改编了这段代码:
def distinct_permutations(seq):
from collections import Counter
def perm_unique_helper(item_counts, perm, i):
if i < 0:
yield tuple(perm)
else:
for item in item_counts:
if item_counts[item] <= 0:
continue
perm[i] = item
item_counts[item] -= 1
# In Python < 3.3 you can replace the yield from with a loop
yield from perm_unique_helper(item_counts, perm, i - 1)
item_counts[item] += 1
item_counts = Counter(seq)
L = len(seq)
return perm_unique_helper(item_counts, [0] * L, L - 1)
我的笔记本电脑无法使用set(permutations(seq))方法执行长度为11的输入序列,但是使用此方法可以!