LeetCode 39 40 組合總和 組合總和II (回溯)
阿新 • • 發佈:2018-11-01
1.組合總和
難度:中等
給定一個無重複元素的陣列 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<int> items; vector<vector<int>> result; backtrade(candidates,0,target,result,items); return result; } void backtrade(vector<int>& candidates, int start, int target, vector<vector<int>> &result, vector<int> items){ if(target<0) return; //剪支 if(target==0) result.push_back(items); for(int i=start;i<candidates.size();i++){ //從start開始,避免重複 items.push_back(candidates[i]); backtrade(candidates,i,target-candidates[i],result,items); items.pop_back(); } } };
2.組合總和II
難度:中等
給定一個數組 candidates 和一個目標數 target ,找出 candidates 中所有可以使數字和為 target 的組合。
candidates 中的每個數字在每個組合中只能使用一次。
說明:
所有數字(包括目標數)都是正整數。
解集不能包含重複的組合。
示例 1:
輸入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集為:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]
示例 2:
輸入: candidates = [2,5,2,1,2], target = 5,
所求解集為:
[
[1,2,2],
[5]
]
思路:其實這題跟上一題差不多,只是要求不重複即可。
class Solution {
public:
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
vector<vector<int>> result;
vector<int> items;
sort(candidates.begin(),candidates.end());
backtrade(candidates,0,target,result,items);
return result;
}
void backtrade(vector<int>& candidates,int start,int target,vector<vector<int>>& result,vector<int> items){
if(target<0) return;
else if(target==0) result.push_back(items);
else {
for(int i=start;i<candidates.size();i++){
if(i>start && candidates[i]==candidates[i-1]) continue;
items.push_back(candidates[i]);
backtrade(candidates,i+1,target-candidates[i],result,items);
items.pop_back();
}
}
}
};