无重复元素的全排列问题
# !/usr/bin/env python
# -*- coding: utf-8 -*-
class Solution:
def permute(self, nums):
if len(path) == len(nums):
result.append(path[:]) # 注意,这里添加path必须以path[:]形式添加,否则加入的会是空列表!!!
return
for j in range(len(nums)):
if vis[j] == 0:
path.append(nums[j])
vis[j] = 1
self.permute(nums)
vis[j] = 0
path.pop()
result = list()
path = list()
nums = [1, 2, 3]
l = len(nums)
vis = [0] * l
p = Solution()
p.permute(nums)
print(result)
优化,不使用vis数组,每次从数组的第i个元素起向后遍历选择一个做为排列的第i个,则此时需要将原数组选中位置与i位置元素互换,才能继续从后面选择。此时也可省去path数组,元素互换后所得原数组即为新的排列。
# !/usr/bin/env python
# -*- coding: utf-8 -*-
class Solution:
def permute(self, nums, i):
if i == len(nums): # 当i来到末尾,则新排列形成
result.append(nums[:]) # 与path[:]一样,需要以nums[:]形式添加!!!
return
for j in range(i, len(nums)): # 新的排列的第i个位置从nums[i...end]任选一个
nums[i], nums[j] = nums[j], nums[i] # 将所选的位置j元素与i元素互换
self.permute(nums, i+1)
nums[i], nums[j] = nums[j], nums[i] # 回溯: 将所选的位置j元素与i元素换回来
result = list()
nums = [1, 2, 3]
p = Solution()
p.permute(nums, 0)
print(result)