代码随想录回溯问题——组合总和III

题目

找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数，并且每种组合中不存在重复的数字。

说明：

所有数字都是正整数。 解集不能包含重复的组合。 示例 1: 输入: k = 3, n = 7 输出: [[1,2,4]]

示例 2: 输入: k = 3, n = 9 输出: [[1,2,6], [1,3,5], [2,3,4]]

思路

重温一下回溯的本质：回溯相当于一个树形结构，回溯等于递归里面嵌套for循环，for循环里面又嵌套着递归

和组合类似，本题的k相当于是树的深度（因为k表示组合的大小，就是从根节点开始到叶子结点的一条路径，可以理解为深度优先搜索），9是树的宽度（因为整个集合就是9个数）。

path相当于树形结构里面的边，用来存储每次选取的元素，sum用来统计path里的元素的总和

剪枝操作：本题中，已选的元素总和如果已经大于n（题目所要求之和）了，再继续往后遍历就没有意义了，可以进行剪枝

java代码如下：

class Solution {
	List<List<Integer>> result = new ArrayList<>();//存放整个结果集合
	List<Integer> path = new ArrayList<>();//存放单个结果
	
	public List<List<Integer>> combinationSum3(int k,int n) {
		dfs(n,k,1,0);
		return result;
	}
	private void dfs(int n, int k, int startIndex, int sum){//n表示目标的和，k表示组合的个数,sum用来统计当前的和
		//剪枝操作
		if(sum > n){
			return;
		}
		if(path.size() == k){
			if(sum == n){
				result.add(new ArrayList<>(path));//相当于组合多了一个限制，判断是否等于给定的和
			}
			return;
		}
		
		//剪枝9 - (k - path.size()) + 1
		for(int i = startIndex;i <= 9 - (k - path.size()) + 1; i++){//因为本题已经确定了最大元素就是9,1-9
			path.add(i);
			sum += i;
			dfs(n, k, i+1, sum);
			//回溯操作
			path.remove(path.size() - 1);
			sum -= i;
		}
	} 
}