【分治算法】【Python实现】排列问题

个人主页：丷从心·

系列专栏：分治算法

学习指南：算法学习指南

---

问题描述

• 设 R = {   r 1 , r 2 , ⋯   , r n   } R = \set{r_{1} , r_{2} , \cdots , r_{n}} R={r1​,r2​,⋯,rn​}是要进行全排列的$n$个元素

---

分治算法

• R i = R − {   r i   } R_{i} = R - \set{r_{i}} Ri​=R−{ri​}

• 集合$X$中的元素的全排列记为$Perm(X)$，$(r_i)Perm(X)$表示在全排列$Perm(X)$的每个排列前加上前缀$r_i$得到的排列

• $R$的全排列可递归定义为

  • 当$n=1$时，$Perm(R)=(r)$，其中$r$是集合$R$中唯一的元素

  • 当$n>1$时，$Perm(R)$由$(r_1)Perm(R_1)$，$(r_2)Perm(R_2)$，$\cdots$，$(r_n)Perm(R_n)$构成

---

Python实现

def permute(nums):
    if len(nums) <= 1:
        return [nums]

    res = []

    for i in range(len(nums)):
        m = nums[i]
        remaining_nums = nums[:i] + nums[i + 1:]

        sub_permutations = permute(remaining_nums)

        for p in sub_permutations:
            res.append([m] + p)

    return res


nums = [1, 2, 3]

permutations = permute(nums)

for p in permutations:
    print(p)

[1, 2, 3]
[1, 3, 2]
[2, 1, 3]
[2, 3, 1]
[3, 1, 2]
[3, 2, 1]

---