Skip to main content

39. 组合总和 [medium]

39. 组合总和 [medium]

https://leetcode-cn.com/problems/combination-sum/

给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的数字可以无限制重复被选取。

说明:

所有数字(包括 target)都是正整数。 解集不能包含重复的组合。 示例 1:

输入:candidates = [2,3,6,7], target = 7,
所求解集为:
[
[7],
[2,2,3]
]

示例 2:

输入:candidates = [2,3,5], target = 8,
所求解集为:
[
[2,2,2,2],
[2,3,3],
[3,5]
]

提示:

  • 1 <= candidates.length <= 30
  • 1 <= candidates[i] <= 200
  • candidate 中的每个元素都是独一无二的。
  • 1 <= target <= 500

通过次数113,615 | 提交次数163,883

First Try

2020-07-24

用递归回溯暴力解法,竟然直接就通过了,但是其实稍微选一下组合多一点的测试案例,直接就超时了,比如candidates = [1, 2, 5, 10, 30], target = 300。总觉得应该有DP缓存之类,时间复杂度感觉也挺难估算出来的。。。

在递归遍历的时候,本来是想着每个元素选择1到n次的,而且按照排序来,后序能选择的只能比当前元素大,不过这样对每个元素都要来一遍。后来发现每个元素按照从0到n次来,直接一波就带走了。

递归遍历每次选择一个较小元素的n次可能往后走,这个写法也能使得选择的元素之间不出现重复,比如target为5可以由2和3来组成,不会出现(2, 3)和(3,2)这两种选择。

class Solution:
def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
# 又是只能递归+暴力解法了吗?
"""
其实用下面的测试案例,直接就超时了,结果竟然能通过,时间还排在50%之前
[1, 2, 5, 10, 30]
300
"""

candidates = sorted(candidates)
if len(candidates) == 0 or candidates[0] > target:
return []

def combo(arr, start, end, target, seq, rv):
if target == 0:
rv.append(seq[:])
if start > end or arr[start] > target:
return
counts, res = divmod(target, arr[start])
for i in reversed(range(0, counts + 1)): # 不使用也算上
seq.extend([arr[start]] * i)
combo(arr, start + 1, end, target - arr[start] * i, seq, rv)
# wtf, 又是一个傻逼bug,在引用的时候修改引用本身; 要么就每次调用前复制一遍
# seq = seq[: -i]
for _ in range(i):
seq.pop()

rv = []
combo(candidates, 0, len(candidates)-1, target, [], rv)
return rv
  • 执行用时:76 ms, 在所有 Python3 提交中击败了52.46%的用户
  • 内存消耗:13.7 MB, 在所有 Python3 提交中击败了5.00%的用户
  • 遇到的bug

另外遇到一个对内存引用理解不深刻的bug,本来以为已经很熟悉了。。

在回溯算法中,列表如果要重复使用,就不能进行引用修改,只能对数组本身进行inplace的修改。 比如seq = seq[:-i]就不能出现,而应该是for _ in range(i): seq:pop().

for i in reversed(range(0, counts + 1)):  # 不使用也算上
seq.extend([arr[start]] * i)
combo(arr, start + 1, end, target - arr[start] * i, seq, rv)
# wtf, 又是一个傻逼bug,在引用的时候修改引用本身; 要么就每次调用前复制一遍
# seq = seq[: -i]
for _ in range(i):
seq.pop()