问题描述
给定一个数组 candidates
和一个目标数 target
,找出 candidates
中所有可以使数字和为 target
的组合。
candidates
中的每个数字在每个组合中只能使用一次。
说明:
- 所有数字(包括目标数)都是正整数。
- 解集不能包含重复的组合。
示例 1:
输入: candidates = [10,1,2,7,6,1,5]
, target = 8
,
所求解集为:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]
示例 2:
输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
[1,2,2],
[5]
]
解题思路
39. 组合总和 相当于完全背包问题,本题相当于多重背包问题,需要考虑每个数字出现的次数。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28 | from collections import Counter
class Solution:
def combinationSum2(
self, candidates: List[int], target: int
) -> List[List[int]]:
count = []
for k, v in Counter(candidates).items():
count.append([k, v])
count.sort(key=lambda x: x[0])
ans = []
def dfs(i: int = 0, tot: int = 0, path: List[int] = []):
if tot == target:
ans.append(path)
return
for j in range(i, len(count)):
if count[j][0] + tot > target:
break
if count[j][1]:
count[j][1] -= 1
dfs(j, tot + count[j][0], path + [count[j][0]])
count[j][1] += 1
dfs()
return ans
|