【DFS】LeetCode 39. 组合总和

题目链接

39. 组合总和

思路

使用 DFS 进行排列组合，首先判断结束递归条件：

• sum > target
• sum == target
• index == len

因为每个数能取无限次，所以从 0 开始慢慢将 candidates[index] 加入到 sum 中，直到 sum > target，并且在循环过程中使用 cnt 记录加入的 candidates[index] 数量。方便循环结束进行弹出。

代码

class Solution {
    private List<List<Integer>> result = new ArrayList<>();
    private Deque<Integer> temp = new ArrayDeque<>();
    private int sum = 0;

    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        dfs(candidates, 0, candidates.length, target);

        return result;
    }

    void dfs(int[] candidates, int index, int len, int target){
        if(sum > target){
            return;
        }
        if(sum == target){
            result.add(new ArrayList<>(temp));
            return;
        }
        if(index == len){
            return;
        }

        int cnt = 0;
        while(sum <= target){
            dfs(candidates, index + 1, len, target);
            sum += candidates[index];
            temp.add(candidates[index]);
            cnt++;
        }
        while(cnt-- > 0){
            sum -= temp.pollLast();
        }
    }
}