LeetCode 46.全排列

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/permutations

给定一个 没有重复 数字的序列,返回其所有可能的全排列。

示例:

输入: [1,2,3]
输出:
[
  [1,2,3],
  [1,3,2],
  [2,1,3],
  [2,3,1],
  [3,1,2],
  [3,2,1]
]

解题思路:首先注意到题目中给定的是没有重复数字的序列,所以我们可以每次都选择序列的第一个数字,然后进行回溯直到序列中没有数据,但是因为python数据结构的特殊性,所以每次进行copy的时候都要使用deepcopy,AC代码如下:

class Solution:
    def permute(self, nums: List[int]) -> List[List[int]]:
        ans = []
        self.calculate(nums, [], ans)
        return ans

    def calculate(self, nums: List[int], temp: List[int], ans: List[List[int]]):
        length = len(nums)
        for i in range(length):
            # 对存储数据的temp列表进行复制
            another = temp[:]
            # 将选中的数据加入到其中
            another.append(nums[i])
            # 将选中的数据弹出
            t = nums[:]
            t.pop(i)
            # 判断是否加入到ans中
            if not t:
                ans.append(another[:])
                return
            self.calculate(t[:], another[:], ans)

最近看了python cookbook,发现python自带了一个获取全排列的模块具体使用如下:

from itertools import permutations

if __name__ == "__main__":
    n = 3
    temp = [i for i in range(n)]
    for p in permutations(temp):
        print(p)

输出结果如下:

LeetCode 46.全排列_第1张图片

你可能感兴趣的:(LeetCode,Python学习)