LeetCode每日一题 216. 组合总和III

216. 组合总和III

Tag Array Backtracking

Difficulty Medium

LInk https://leetcode-cn.com/problems/combination-sum-iii/

分析

回溯+剪枝

见代码注释

题解

class Solution {
    List<List<Integer>> list = new ArrayList<>();
    Stack<Integer> path = new Stack<>();
    
    public List<List<Integer>> combinationSum3(int k, int n) {
        dfs(k, n, 1);
        return list;
    }
    
    private void dfs(int k, int n, int start) {
      // 匹配失败，剪枝
        if (k<0 || n<0) {
            return;
        }
      // 如果符合条件，将组合加入结果中
        if (k==0 && n==0) {
            if (!path.isEmpty()) {
                list.add(new ArrayList<>(path));
            }
            return;
        }
      // 从1-9，递归，注意退回后要恢复状态
        for (int i=start; i<10; i++) {
            path.push(i);
            dfs(k-1, n-i, ++start);
            path.pop();
        }
    }
}