LeetCode——组合总和(Combination Sum)

题目:

给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的数字可以无限制重复被选取。

说明:

  • 所有数字(包括 target)都是正整数。
  • 解集不能包含重复的组合。 

示例 1:

输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
  [7],
  [2,2,3]
]

示例 2:

输入: candidates = [2,3,5], target = 8,
所求解集为:
[
  [2,2,2,2],
  [2,3,3],
  [3,5]
]

思路

首先注意列表中的数字是从小到大排列,其次每个数可以无限取。这种组合求和的题目,首先想到的就是深度遍历。

LeetCode——组合总和(Combination Sum)_第1张图片

思路跟上图一样,就是一直深度遍历,假如求和结果大于target则返回,等于则添加到结果集中,小于则继续深度遍历。

代码:

class Solution:
    def combinationSum(self, candidates, target):
        """
        :type candidates: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        self.result = []#存放结果集
        self.target1 = target
        for index, num in enumerate(candidates):
            temp = []  # 保存一次计算的结果
            temp.append(num)
            self.helper(candidates, index, temp)
        return self.result
    def helper(self,candidates,index,temp):
        temp1 = copy.deepcopy(temp)
        if sum(temp1) == self.target1:
            self.result.append(temp1)
            return
        elif sum(temp1) < self.target1:
            for i ,num in enumerate(candidates[index:]):
                temp1.append(num)
                self.helper(candidates,index+i,temp1)
                temp1.pop()
        elif sum(temp1) > self.target1:
             return
if __name__ =="__main__":
    res = Solution()
    print(res.combinationSum([2, 3, 5], 8))
    pass

 

你可能感兴趣的:(算法题)