leetcode —— 39. 组合总和

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

说明:

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

示例 1:

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

解题思路:使用递归+回溯,为了避免组合出现重复,例如示例1中的重复组合[2,2,3]和[3,2,2],可以先对输入candidates进行排序,然后在递归的时候,设temp为当前组合,下一个可以放进temp中的数,一定大于等于temp[-1]。

其Python3代码如下:

# Python3
class Solution:
    def __init__(self):
        self.ans = []
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        if not candidates:
            return []
        def combSums(candidates,target,temp,sums):
            if target == sums:  # 递归停止条件
                self.ans.append(temp[:])
                return

            for n in candidates:  # 遍历每种情况
                if sums + n > target:  # 如果当前sums值大于target,直接break
                    break
                if temp and n < temp[-1]:  # 如果当前输入的值小于temp[-1]则continue
                    continue
                temp.append(n)
                combSums(candidates,target,temp,sums+n)
                temp.pop()  # 回溯
        candidates.sort()  # 执行递归时先进行排序
        combSums(candidates,target,[],0)
        return self.ans

其C++代码如下:

# C++
class Solution {
private:
    vector<vector<int>> num;
    vector<int> sum;
    int i = 0;
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) 
    {
        sort(candidates.begin(),candidates.end());
        function(candidates,target,sum,i);
        return num;
    }
    void function(vector<int>& candidate, int target,vector<int> sum,int i)
    {
        if(target==0)
        {
            num.push_back(sum);
        }        
        for(;i<candidate.size();++i)  # 只遍历比当前数字大的candidates数字
        {
            if(candidate[i]<=target)
            {
                sum.push_back(candidate[i]); 
                function(candidate,target-candidate[i],sum,i);
                sum.pop_back();
            }
        }
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章