跳转至

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. 组合总和 相当于完全背包问题,本题相当于多重背包问题,需要考虑每个数字出现的次数。


 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
返回顶部

在手机上阅读