LeetCode-Python-39. 组合总和

给定一个无重复元素的数组 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]
]

 

思路:

回溯法。

#自己写的丑陋版本,很慢……
class Solution(object):
    def combinationSum(self, candidates, target):
        """
        :type candidates: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        
        res = list()
        
        def generate(c, t, tmp, s):
            # print s
            
            if s == target:
                # print tmp
                res.append(tmp[:])
                return
            if s > t:
                return
            
            for digit in c:
                s = sum(tmp) + digit
                tmp.append(digit)
                generate(c, t, tmp, s)
                tmp.pop()
                
        generate(candidates, target, [], 0)
        #----以下为去重
        for i in range(0, len(res)):
            res[i].sort()
        ress = list()
        
        for i in range(0, len(res)):
            flag = 0 # 1 重复 0 单独
            for j in range(i + 1, len(res)):
                if res[i] == res[j]:
                    flag = 1
                    break
            if not flag:
                ress.append(res[i])
                
        return ress
                
#大神写的
class Solution(object):
    def combinationSum(self, candidates, target):
        res = []
        candidates.sort()
        
        def backtrack(remain, temp, start):
            if not remain: #remain为0
                res.append(temp[:])
            else:
                for i, n in enumerate(candidates[start:]):
                    if n > remain:
                        break
                    backtrack(remain-n, temp+[n], start+i)
        backtrack(target, [], 0)
        return res

你可能感兴趣的:(Leetcode,Python)