仅供自己学习记录
LeetCode刷题日记2021-3-31
解题思路回溯法用递归实现
题目描述
给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。
示例 1:
输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]
示例 2:
输入:nums = [0]
输出:[[],[0]]
解题思路:
回溯法是一种类似于枚举的解题思路 在搜索的过程中发现不满足解题条件就回溯,尝试别的路径
回溯法的整体思路是:搜索每一条路,每次回溯是对具体的一条路径而言的。对当前搜索路径下的的未探索区域进行搜索,则可能有两种情况:
1.当前未搜索区域满足结束条件,则保存当前路径并退出当前搜索;
2.当前未搜索区域需要继续搜索,则遍历当前所有可能的选择:如果该选择符合要求,则把当前选择加入当前的搜索路径中,并继续搜索新的未探索区域。
一般的回溯模板如下
res = []
path = []
def backtrack(未探索区域, res, path):
if path 满足条件:
res.add(path) # 深度拷贝
# return # 如果不用继续搜索需要 return
for 选择 in 未探索区域当前可能的选择:
if 当前选择符合要求:
path.add(当前选择)
backtrack(新的未探索区域, res, path)
path.pop()
backtrack 的含义是:未探索区域中到达结束条件的所有可能路径,path 变量是保存的是一条路径,res 变量保存的是所有搜索到的路径。所以当「未探索区域满足结束条件」时,需要把 path 放到结果 res 中。
path.pop() 是啥意思呢?它是编程实现上的一个要求,即我们从始至终只用了一个变量 path,所以当对 path 增加一个选择并 backtrack 之后,需要清除当前的选择,防止影响其他路径的搜索。
题解代码:
def subsetsWithDup(self, nums: List[int]) -> List[List[int]]:
res,path=[],[]
nums.sort()
self.dfs(nums,0,res,path)
return res
def dfs(self,nums,index,res,path):
res.append(copy.deepcopy(path))
for i in range(index,len(nums)):
#来判断去重
if i>index and nums[i]==nums[i-1]:
continue
path.append(nums[i])
self.dfs(nums,i+1,res,path)
path.pop()
来源:
https://leetcode-cn.com/problems/subsets-ii/solution/hui-su-fa-mo-ban-tao-lu-jian-hua-xie-fa-y4evs/