leetcode hot 100搜索回溯

39. 组合总和

已解答

中等

相关标签

相关企业

给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ，找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ，并以列表形式返回。你可以按 任意顺序 返回这些组合。

candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同，则两种组合是不同的。 

对于给定的输入，保证和为 target 的不同组合数少于 150 个。

import copy

class Solution(object):

    def combinationSum(self, candidates, target):

        """

        :type candidates: List[int]

        :type target: int

        :rtype: List[List[int]]

        """

        # 当成一棵树来遍历回溯条件是大于target

        self.ret = []

        def dfs(target , current,current_list,candidates):

            if current==target:

                self.ret.append(copy.deepcopy(current_list))

            elif current>target:

                return

            else:

                for index,i in enumerate(candidates):

                    current+=i

                    current_list.append(i)

                    dfs(target,current,current_list,candidates[index:])

                    del current_list[-1]

                    current-=i

 

        current = 0

        current_list = []

        dfs(target , current,current_list,candidates)

           

        return self.ret

       

需要注意的是，candidates需要每次只能搜索到后面的dfs(target,current,current_list,candidates[index:])