LeetCode 39. 组合总和

原题地址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]]

---

算法

迭代，先对candidates从小到大排序，每次用target减去candidates[i]，如果结果刚好为0，就输出；如果结果小于0，说明已经溢出了，结束；如果结果大于0，那么将candidates[i]记录进结果，并更新target，以及candidates更改为不小于candidates[i]的部分，保证选出的结果不重复。

---

代码

class Solution:
    def combinationSum(self, candidates, target):
        """
        :type candidates: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        candidates.sort()

        def back(lst, tg, res):
            for i in range(len(lst)):
                if tg - lst[i] == 0:
                    yield res + [lst[i]]
                if tg - lst[i] > 0:
                    yield from back(lst[i:], tg - lst[i], res + [lst[i]])
                if tg - lst[i] < 0:
                    break

        return list(back(candidates, target, []))