leetcode_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]
]

二、解题思路

DFS+回溯,去重则排序查找是否重复即可

三、代码

class Solution:
    def combinationSum2(self, candidates: list, target: int) -> list:
        n = len(candidates)
        ans = []
        res = []
        # candidates.sort()
        def dfs(index, n, su, ans):
            if su == target:
                res.sort()
                if res not in ans:
                    ans.append(res.copy())
                # print(ans)
            if su > target:
                return
            for i in range(index, n):
                su += candidates[i]
                res.append(candidates[i])
                dfs(i + 1, n, su, ans)
                su -= candidates[i]
                res.remove(candidates[i])

        dfs(0, n, 0, ans)
        return ans

if __name__ == '__main__':
    candidates = [10,1,2,7,6,1,5]
    target = 8
    s = Solution()
    ans = s.combinationSum2(candidates, target)
    print(ans)

你可能感兴趣的:(leetcode,Python,leetcode,dfs,python,算法,回溯)