LeetCode - 組合總和

題目鏈接:https://leetcode-cn.com/problems/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]
]

我的思路

  • 經典的回溯法,直接套公式即可
class Solution {
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        vector<vector<int>> ans;
        if(candidates.empty()) return ans;
        vector<int> res;
        backtrack(candidates, target, 0, res, ans);
        return ans;
    }
    void backtrack(vector<int>& candidates, int target, int index, vector<int> res, vector<vector<int>>& ans){
        if(target == 0){
            ans.push_back(res);
            return;
        }
        if(target < 0) return;	
        for(int i = index; i < candidates.size(); i++){
            res.push_back(candidates[i]);
            backtrack(candidates, target - candidates[i], i, res, ans);
            res.pop_back();
        }
        return;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章