78. Subsets
题目描述
没有重复数字的全组合
例子
Input: nums = [1,2,3]
Output:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]
解法
法1 - 常规DFS
法2 - 非递归,可以看出是在上一次结果的基础上加入新元素
解法1
递归DFS
class Solution(object):
def subsets(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
res = [[]]
self.dfs(nums, [], res)
return res
def dfs(self, nums, temp, res):
if not nums:
return
for i in range(len(nums)):
temp.append(nums[i])
res.append(temp[:])
self.dfs(nums[i+1:], temp, res)
temp.pop()
解法2
非递归
class Solution(object):
def subsets(self, nums):
res = [[]]
for num in nums:
res += [temp + [num] for temp in res]
return res
90. Subsets II
题目描述
包含重复数字的整数集-全组合
例子
Input: [1,2,2]
Output:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
解法
法1 - DFS,在上题基础上将nums排序。然后判断当前元素nums[i]是否和nums[i-1]相等,若相等,跳出此次循环继续下一次循环。
比如,第一次DFS到尽头后为 [], [1], [1,2], [1,2,2];回溯时-[1,2]pop出2后,[1]尝试与最后一个2组合(重复了2==2),continue
法2 - 非递归。类似于层次添加元素[] → [1] → [2], [1, 2] → [1,2], [2,2]…
若当前元素和上一个元素相等,则本轮 应该从[添加上一个元素]之后的位置添加该元素。
解法1
i是标明开头元素是哪个,然后依次和i以后的元素进行组合
class Solution(object):
def subsetsWithDup(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
nums.sort()
res = [[]]
self.dfs(nums, [], res)
return res
def dfs(self, nums, temp, res):
if not nums:
return
for i in range(len(nums)):
if i > 0 and nums[i] == nums[i-1]:
continue
temp.append(nums[i])
res.append(temp[:])
self.dfs(nums[i+1:], temp, res)
temp.pop()
精简代码
class Solution(object):
def subsetsWithDup(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
nums.sort()
res = [[]]
self.dfs(nums, [], res)
return res
def dfs(self, nums, path, res):
for i in range(len(nums)):
if i > 0 and nums[i] == nums[i-1]:
continue
self.dfs(nums[i+1:], path + [nums[i]], res)
res.append(path + [nums[i]])
解法2
非递归。若当前元素和上一个元素相等,则本轮 应该从[添加上一个元素]之后的位置添加该元素。
class Solution(object):
def subsetsWithDup(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
nums.sort()
res = [[]]
for i in range(len(nums)):
if i == 0 or nums[i] != nums[i-1]:
l = len(res)
for path in res[len(res)-l:]:
res.append(path + [nums[i]])
return res