算法学习：216.组合总和 III

组合总和 III

题目链接：力扣题目链接
难度：中等
找出所有相加之和为 n 的 k 个数的组合，且满足下列条件：


只使用数字1到9
每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。


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

思路

在[1,2,3,4,5,6,7,8,9]这个集合中找到和为n的k个数的组合。
相对比与77.组合，多了一个条件，找到和为n的k个数的组合，而集合[1,2,3,4,5,6,7,8,9]是固定的。
如果把这个过程看做是一个树，则k就相当于树的深度，整个集合就相当于树的宽度。

回溯代码

//找出所有相加之和为 n 的 k 个数的组合
class Solution{
	List<List<Integer>> result = new ArrayList<>(); // 存放结果集
	LinkedList<Integer> path = new LinkedList<>(); // 符合条件的结果
	public List<List<Integer>> combinationSum3(int k, int n) {
		combinationSum3(k,n,1,0);
		return result;
	}
	private void combinationSum3(int k,int n,int startIndex,int sum){
		if(sum > n){// 剪枝操作
            return; // 如果path.size() == k 但sum != targetSum 直接返回
		}
		if(path.size() == k){
			if(sum == n){
				result.add(new ArrayList<>(path));	
			}
			return;
		}
		// 剪枝
		for(int i = startIndex ; i <= 9 - (k - path.size()) + 1 ; i++){
			// 处理
			sum += i; 
			path.add(i);
			combinationSum3(k,n,i + 1,sum);
			// 回溯
			sum -= i;
			path.removeLast();
		}
	}
}