题目
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
代码:
class Solution {
public List<List<Integer>> combine(int n, int k) {
List<List<Integer>> res=new ArrayList<>();
backtrack(res,n,k,1,new ArrayList<Integer>());
return res;
}
//startIndex是用来控制本次递归中从哪开始遍历(因为每次从集合中选取元素,可选择的范围随着选择而收缩)
public void backtrack(List<List<Integer>> res,int n, int k,int startIndex,ArrayList<Integer> tmpList){
//递归出口
if(tmpList.size()==k){
res.add(new ArrayList<>(tmpList));
return;
}
//遍历
for(int i=startIndex;i<=n;i++){
//剪枝条件(如果剩下可选择的元素数量不够组成k个数,就剪枝)
if(i>(n-(k-tmpList.size())+1)){//n-(k-tmpList.size())+1表示最多只能从这个数开始才能构成k个元素
break;
}
tmpList.add(i);
//递归
backtrack(res,n,k,i+1,tmpList);
//回溯
tmpList.remove(tmpList.size()-1);
}
}
}
变量说明:
startIndex:组合是无序的,不同顺序的排列算一种,所以我们可以选择的元素会随着我们的选择而减少,这时需要一个变量来控制我们下一次递归的从哪开始遍历。
递归、回溯、剪枝常伴随,组合、排列都是在叶子节点上收集数据,子集是在每一个节点上收集数据。剪枝就是为了减去已知的不符合条件的路径,省的折腾无用功,通常要求结果集合不能重复,每个元素只能用一次之类的都需要剪枝(判断索引及该元素和之前元素是否重复,跳出for循环)
看了大佬的文章,对回溯更加了解,特此记录,方便日后学习
代码随想录