LeetCode-46 Permutations

Description

Given a collection of distinct integers, return all possible permutations.

Example

Input: [1,2,3]
Output:
[
[1,2,3],
[1,3,2],
[2,1,3],
[2,3,1],
[3,1,2],
[3,2,1]
]

Submissions

这道题要求是返回给定列表所有可能的全排列组合，解题思路是直接采用递归，我们依次遍历nums,固定当前遍历的nums[i],去全排列剩余数字序列，最后将nums[i]分别插入到每个排列中合并即可。

实现代码如下：

class Solution:
    def permute(self, nums: List[int]) -> List[List[int]]:
        res=[]
        if len(nums)<=1:
            return [nums]
        for i in range(len(nums)):
            p=self.permute(nums[:i]+nums[i+1:])
            for j in p:
                res.append([nums[i]]+j)
        return res

同时我还查到一个简单的方法，使用python自带的itertools.permutations()。
itertools.permutations(iterable[, r])指创建一个迭代器，返回iterable中所有长度为r的项目序列，r 指定生成排列的元素的长度，如果不指定，则默认为可迭代对象的元素长度。

实现代码如下：

class Solution:
    def permute(self, nums: List[int]) -> List[List[int]]:
        return [list(i) for i in itertools.permutations(nums)]

参考