python通过yield实现数组全排列

2022-05-03 00:00:00 python 数组 排列

从n个不同元素中任取m(m≤n)个元素,按照一定的顺序排列起来,叫做从n个不同元素中取出m个元素的一个排列。当m=n时所有的排列情况叫全排列。
这段代码用到了yield方法,全排列速度加倍

"""
皮蛋编程(https://www.pidancode.com)
创建日期:2022/3/30
功能描述:python通过yield实现数组全排列
"""


def perm(arr, pos=0):
    if pos == len(arr):
        yield arr
    for i in range(pos, len(arr)):
        arr[pos], arr[i] = arr[i], arr[pos]
        for _ in perm(arr, pos + 1): yield _
        arr[pos], arr[i] = arr[i], arr[pos]


for i in perm([1, 2, 3]):
    print(i)

相关文章