Java | Leetcode Java题解之第216题组合总和III

题目：

题解：

class Solution {
    private List<List<Integer>> res = new ArrayList<>();    // 结果数组

    public List<List<Integer>> combinationSum3(int k, int n) {
        if(n < (k + 1) * k / 2 || n > (19- k) * k / 2)return res;   // n如果小于k能构成的最小值或者大于k能构成的最大值，直接返回
        List<Integer> ans = new ArrayList<>();    // 存储每一种组合
        backtracking(k, n, ans);    // 回溯法
        return res;
    }
        
    /**
     * 回溯法，对于当前k和n, 枚举元素
     */
    private void backtracking(int k, int n, List<Integer> ans){
        if(k == 0 || n <= 0){
            // k == 0 表示已经使用了k个数，n <= 0表示这k个数元素和已经大于等于n，这些情况无需再枚举元素了
            if(k == 0 && n == 0){
                // 使用k个数且恰好等于n，加入结果res
                res.add(new ArrayList<>(ans));
            }
            return;
        }
        // 枚举起点，等于当前数组中最后一个元素+1，数组为空相当于起始为1
        int start = (ans.size() == 0 ? 0 : ans.get(ans.size() - 1)) + 1;
        for(int i = start; i < 10; i++){
            if(n - i < 0)break;  // 加上元素i已经超过n，更大的i也就无须枚举了
            ans.add(i);    // 加入枚举的数
            backtracking(k - 1, n - i, ans);    // 递归搜索，k-1表示使用了一个数，n-i表示使用了i元素
            ans.remove(ans.size() - 1);         // 弹出加入的数，还原现场
        }
    }
}