原题地址40. 组合总和 II
给定一个数组 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. 组合总和。区别就在于判断一些相同数字的使用次数。
代码
class Solution:
def combinationSum2(self, candidates, target):
"""
:type candidates: List[int]
:type target: int
:rtype: List[List[int]]
"""
import collections
dst = collections.Counter(candidates)
candidates = list(dst.keys())
candidates.sort()
def back(lst, tg, res):
for i in range(len(lst)):
j = 1
while j <= dst[lst[i]]:
if tg - j * lst[i] == 0:
yield res + j * [lst[i]]
if tg - j * lst[i] > 0:
yield from back(lst[(i + 1):], tg - j * lst[i], res + j * [lst[i]])
if tg - j * lst[i] < 0:
break
j += 1
return list(back(candidates, target, []))