我正在尝试找到一种有效的算法来查找多重集的排列,给定索引。
Ex:给定{1, 3, 3}
。升序词典顺序中的所有排列均为{133, 313, 331}
。这些元素的索引为{0, 1, 2}
。给定index=2
,结果为331。
I found an algorithm在给定词典索引的情况下查找集合的排列。他的算法是有效的:O(n ^ 2)。
但是,算法在适当的集合(例如{1, 2, 3}
)上进行测试,并且在我的测试中不正确。我在这里描述了他的python代码,以便您可以轻松地遵循。
from math import factorial, floor #// python library
from math import factorial, floor #// python library
i=5 #// i is the lexicographic index (counting starts from 0)
n=3 #// n is the length of the permutation
p = range(1,n+1) #// p is a list from 1 to n
for k in range(1,n+1): #// k goes from 1 to n
d = i//factorial(n-k) #// use integer division (like division+floor)
print(p[d]),
p.remove(p[d]) #//delete p[d] from p
i = i % factorial(n-k) #// reduce i to its remainder
答案 0 :(得分:4)
# Python 2
from collections import Counter
from math import factorial
def count_permutations(counter):
values = counter.values()
return (
factorial(sum(values))/reduce(lambda a, v: a * factorial(v), values, 1)
)
def permutation(l, index):
l = sorted(l)
if not index:
return l
counter = Counter(l)
total_count = count_permutations(counter)
acc = 0
for i, v in enumerate(l):
if i > 0 and v == l[i-1]:
continue
count = total_count * counter[v] / len(l)
if acc + count > index:
return [v] + permutation(l[:i] + l[i + 1:], index - acc)
acc += count
raise ValueError("Not enough permutations")
似乎按预期工作
In [17]: for x in range(50): print x, permutation([1, 1, 2, 2, 2], x)
0 [1, 1, 2, 2, 2]
1 [1, 2, 1, 2, 2]
2 [1, 2, 2, 1, 2]
3 [1, 2, 2, 2, 1]
4 [2, 1, 1, 2, 2]
5 [2, 1, 2, 1, 2]
6 [2, 1, 2, 2, 1]
7 [2, 2, 1, 1, 2]
8 [2, 2, 1, 2, 1]
9 [2, 2, 2, 1, 1]
10---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
[...]
ValueError: Not enough permutations
时间复杂度:O(n^2)
。