LeetCode–組合總和

部落格說明

文章所涉及的資料來自網際網路整理和個人總結，意在於個人學習和經驗彙總，如有什麼地方侵權，請聯絡本人刪除，謝謝！

介紹

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]
]

提示：

1 <= candidates.length <= 30
1 <= candidates[i] <= 200
candidate 中的每個元素都是獨一無二的。
1 <= target <= 500

思路

回溯演算法 + 剪枝

• 輸入: candidates = [2, 3, 6, 7]，target = 7。
• 候選數組裡有 2，如果找到了組合總和為 7 - 2 = 5 的所有組合，再在之前加上 2 ，就是 7 的所有組合；
• 同理考慮 3，如果找到了組合總和為 7 - 3 = 4 的所有組合，再在之前加上 3 ，就是 7 的所有組合，依次這樣找下去。

程式碼

class Solution {
    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        int len = candidates.length;
        List<List<Integer>> res = new ArrayList<>();
        if(len == 0){
            return res;
        }
        Deque<Integer> path = new ArrayDeque<>();
        dfs(candidates,0,len,target,path,res);
        return res;
    }

    public void dfs(int[] candidates,int begin,int len,int target,Deque<Integer> path,List<List<Integer>> res){
        if(target < 0){
            return;
        }
        if(target == 0){
            res.add(new ArrayList<>(path));
        }
        for(int i = begin; i < len; i++){
            path.addLast(candidates[i]);
            dfs(candidates,i,len,target-candidates[i],path,res);
            path.removeLast();
        }
    }
}
 

感謝

Leetcode

以及勤勞的自己，個人部落格，GitHub