代码随想录回溯算法——组合问题

题目

给定两个整数 n 和 k，返回 1 … n 中所有可能的 k 个数的组合。

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

思路

回溯法三部曲：

1. 递归函数的返回值以及参数
2. 回溯函数终止条件
3. 单层搜索 （递归）的过程（回溯法的搜索过程是一个树形结构的遍历过程，for循环用来横向遍历，递归的过程是纵向遍历）

回溯法模板框架：（回溯相当于递归里面嵌套着for循环，for循环里面又继续嵌套着递归）

void backtracking(参数){
	if(终止条件){
		存放结果;
		return;
	}
	for(选择：本层集合中元素（树中节点的孩子数量就是集合的大小）){
		处理节点;
		backtracking(路径，选择列表);//递归
		回溯，撤销处理结果;
	}
}

相当于是树形结构上的深度优先遍历：
1.如果组合里有1，那么需要在[2，3，4]里找一个数
2.如果组合里有2，那么需要在[3，4]里找一个数，注意这里不能再找1了，因为包含1的组合已经在第一种情况中包含了

java代码如下：

class Solution {
	List<List<Integer>> result = new ArrayList<>();//存放最终的结果
	List<Integer> path = new ArrayList<>();//存放每个结果，并且充当栈的作用，进行回溯操作
	public List<List<Integer>> combine(int n, int k){
		dfs(n,k,1);
		return result;
	}
	//每次从集合中选取元素，可选择的范围随着选择的进行而收缩，调整可选择的范围，就是要靠startIndex
	//@param startIndex 用来记录本层递归的中，集合从哪里开始遍历（集合就是[1,...,n] ）
	private void dfs(int n,int k,int startIndex){
		//终止条件
		if(path.size() == k){//因为一次是要找k个数的组合
			result.add(new ArrayList<>(path));
			return;
		}
		//单层逻辑
		for(int i = startIndex; i <= n-(k-path.size()) + 1; i++){
		//剪枝过程：已经选择的元素为path.size(),还需要的元素个数为k-path.size(),剪枝是因为从i后面开始，不能组成长度为k的组合（长度不足）
		//比如，如果 n = 7, k = 4，从 5 开始搜索就已经没有意义了，这是因为：即使把 5 选上，后面的数只有 6 和 7，一共就 3 个候选数，凑不出 4 个数的组合
			path.add(i);
			dfs(n,k,i+1);
			path.remove(path.size() - 1);//removeLast()方法是LinkedList的，如果是ArrayList那只能这么写
		}
	}
}