LC 216. 组合总和 III

题干

找出所有相加之和为 n **的 k ****个数的组合，且满足下列条件：

• 只使用数字1到9
• 每个数字 最多使用一次 

返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7
没有其他符合的组合了。

思路

主要考察点：回溯

• 难点：怎么限制层数和总和
• 使用sum记录总和，k记录层数
• 终结条件：只要在层数k=0时收集结果或结束回溯，K<0直接被截断

解题

class Solution {
    List<List<Integer>> res = new ArrayList<List<Integer>>();
    List<Integer> path = new ArrayList<Integer>();
    public List<List<Integer>> combinationSum3(int k, int n) {
        backtrace(n,k,0,1);
        return res;
    }
    void backtrace(int n,int k,int sum,int startIndex){
        if(sum==n && k==0){
            res.add(new ArrayList<Integer>(path));
            return;
        }else if(k==0){
            return;
        }
        for(int i=startIndex;i<=9;i++){
            path.add(i);
            sum+=i;
            backtrace(n,k-1,sum,i+1);
            path.remove(path.size()-1);
            sum-=i;
        }
        return;
    }
}