python 求排列组合

Python中有多种方式可以求排列组合，下面介绍两种常见的方法。

1. 使用itertools模块

Python的itertools模块提供了多种排列组合的函数，包括permutations（排列）和combinations（组合）。这些函数可以快速方便地生成排列组合结果。

例如，要求长度为3的列表[1,2,3]的所有排列，可以使用itertools.permutations函数：

import itertools

lst = [1,2,3]
perms = itertools.permutations(lst, 3)
for perm in perms:
    print(perm)

输出结果为：

(1, 2, 3)
(1, 3, 2)
(2, 1, 3)
(2, 3, 1)
(3, 1, 2)
(3, 2, 1)

要求长度为2的列表[1,2,3]的所有组合，可以使用itertools.combinations函数：

import itertools

lst = [1,2,3]
combs = itertools.combinations(lst, 2)
for comb in combs:
    print(comb)

输出结果为：

(1, 2)
(1, 3)
(2, 3)

2. 自己实现排列组合函数

如果不想使用itertools模块，也可以自己实现排列组合函数。下面是一个简单的实现：

def permutations(lst, n):
    if n == 0:
        yield []
    else:
        for i in range(len(lst)):
            for perm in permutations(lst[:i] + lst[i+1:], n-1):
                yield [lst[i]] + perm

def combinations(lst, n):
    if n == 0:
        yield []
    else:
        for i in range(len(lst)-n+1):
            for comb in combinations(lst[i+1:], n-1):
                yield [lst[i]] + comb

这里的permutations函数和combinations函数分别实现了排列和组合。它们都使用了递归算法，对于每个元素，可以选择使用或者不使用。当选择了n个元素后，即得到一个排列或组合。