Python 生成所有非递减序列

Posted 2023-02-25

【中文标题】Python 生成所有非递减序列

【英文标题】：Python generating all nondecreasing sequences

【发布时间】：2015-07-22 00:52:15

【问题描述】：

我很难找到一种以 Python 方式执行此操作的方法。我假设我可以以某种方式使用 itertools，因为我以前做过类似的事情，但不记得我做了什么。

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

【参考方案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 保留了输入的顺序，而且由于我们传入的是非递减序列，因此我们只会得到非递减元组。

（如果您确实需要列表而不是元组，则很容易转换为列表。）