Python生成所有非递减序列

Question

我很难找到一种以 Python 方式做到这一点的方法。我认为我可以以某种方式使用 itertools，因为我以前做过类似的事情，但不记得我做了什么。

我正在尝试生成长度为 L 的所有非递减列表，其中每个元素可以采用 1 到 N 之间的值。例如，如果 L=3 且 N=3 则 [1,1,1],[1 ,1,2],[1,1,3],[1,2,2],[1,2,3]等

Answer 1

您可以使用 itertools.combinations_with_replacement 来执行此操作:

>>> L, N = 3,3
>>> cc = combinations_with_replacement(range(1, N+1), L)
>>> for c in cc: print(c)
(1, 1, 1)
(1, 1, 2)
(1, 1, 3)
(1, 2, 2)
(1, 2, 3)
(1, 3, 3)
(2, 2, 2)
(2, 2, 3)
(2, 3, 3)
(3, 3, 3)

这是有效的，因为 c_w_r 保留了输入的顺序，并且由于我们传递的是非递减序列，所以我们只能得到非递减元组。

(如果您确实需要列表而不是元组，则可以轻松转换为列表。)