LeetCode演算法題40:組合總和 II解析
阿新 • • 發佈:2019-01-03
給定一個數組 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] ]
這個題和組合總和第一個題基本一樣,就是要在搜尋下一個數的時候不要從本次這個數開始,而是從下一個數開始,也就是遞迴的時候傳遞的值變為i+1。而且有可能會有重複的數,所以for迴圈的時候跳過重複的數。
C++原始碼:
class Solution {
public:
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
vector<vector<int>> res;
vector< int> out;
sort(candidates.begin(), candidates.end());
combinationSum2DFS(candidates, target, 0, out, res);
return res;
}
void combinationSum2DFS(vector<int>& candidates, int target, int start, vector<int>& out, vector<vector<int>>& res) {
if(target < 0) return;
else if(target == 0) {res.push_back(out); return;}
for(int i=start;i<candidates.size();i++){
if(i>start && candidates[i]==candidates[i-1]) continue;
out.push_back(candidates[i]);
combinationSum2DFS(candidates, target-candidates[i], i+1, out, res);
out.pop_back();
}
}
};
python3原始碼:
class Solution:
def combinationSum2(self, candidates, target):
"""
:type candidates: List[int]
:type target: int
:rtype: List[List[int]]
"""
res = []
out = []
candidates.sort()
self.combinationSum2DFS(candidates, target, 0, out, res)
return res
def combinationSum2DFS(self, candidates, target, start, out, res):
if target < 0:
return
elif target == 0:
res.append(copy.deepcopy(out))
return
for i in range(start, len(candidates)):
if i>start and candidates[i]==candidates[i-1]:
continue
out.append(candidates[i])
self.combinationSum2DFS(candidates, target-candidates[i], i+1, out, res)
out.pop()