您的位置:首页 > 编程语言 > Python开发

leetcode 46 Permutations Python 实现(回溯算法)

2019-06-01 17:52 2351 查看

 

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]
]

回溯算法自己的一个思路如下:需要保留数字使用情况,my_nums增加和删除元素等操作
class Solution:
def permute(self, nums: List[int]) -> List[List[int]]:
results = []
use_dict = dict.fromkeys(nums, False) #初始化一个字典,保存数字是否使用过
def backtrack(results, my_nums, use_dict):
if len(my_nums) == len(nums):
tmp_nums = copy.deepcopy(my_nums)
results.append(tmp_nums)
return

for x in nums:
if not use_dict[x]:
use_dict[x] = True
my_nums.append(x)
backtrack(results, my_nums, use_dict)
my_nums.remove(x)  #remove操作耗费时间
use_dict[x] = False
backtrack(results, [], use_dict)
return results

整体程序花费时间较多,remove处消耗太多时间, 超过24%左右

 改良版:

class Solution:
def permute(self, nums: List[int]) -> List[List[int]]:
results = []
len_n = len(nums)
def backtrack(my_nums, use_nums):
if len(my_nums) == len_n:
results.append(my_nums)
return
for i in range(len(use_nums)):
x = my_nums.copy()
x.append(use_nums[i])
backtrack(x, use_nums[:i]+use_nums[i+1:])
backtrack([], nums)
return results

这个每次递归会缩小剩下可用数字范围,不需要维护数字使用dict

my_nums每次都copy一份,这样就不需要remove操作了,运行时间缩短。 

超过98%以上Python速度

 

 

---最近转战英文leetcode站了,所以题目变成了英文描述
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: