来源:力扣(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)
输出结果如下: