首先,不用回溯的方法,我们就得不断的for循环,k有多少就得循环多少,这样显然不行的,所以像遇到这种情况,就得使用回溯了。
算法的过程图如下所示,黄色箭头表示回溯操作,第二层方框中的竖线前面表示已选,后面表示待选。
现在我们一步一步去写代码,回溯问题有模板的,比如这篇文章就写的很好,我们在这里就用这种方法。回溯模板参考文章
//首先,声明变量
//需要声明一个存放最终结果的List
List<List<Integer>> result = new ArrayList<>();
//声明一个存放路径的数组,用ArrayDeque是因为它能够直接删除最后一个
Deque<Integer> path = new ArrayDeque<>();
//然后,写回溯函数
void backTrack(int n, int k, int startIndex);
//第一,写终止条件
if(path.size() == k){//从上图能够看到,当path遍历深度为k的时候该返回结果了
result.add(new ArrayList<>(path));//将本次结果加入最终结果中
return;
}
//进行上面的绿色for循环
for(int i = startIndex; i <= n; ++i){
path.addLast(i);//先将元素加进去
backTrack(n,k, i + 1);
path.removeLast(i);//回溯操作,将上面添加的元素删了,比如黄色线
}
//这里要重点说一下这个startIndex
//它的作用就是为了保证没有重复组合,比如[1,2],[2,1]就算是重复;
//比如,第一层肯定从索引0开始遍历,当选中了a后,下层只能从索引1开始遍历
//当第一层选中了b后,它的下一层只能从索引2开始遍历,就如同上图的标识的那样
//这样从广度方向来说,你可一观察这二层的框,i的起始都不一样,为startIndex;
//从深度方向来说,下一层的起始在上一层选中的索引加1
所以综合以上,代码如下:
class Solution {
Deque<Integer> path = new ArrayDeque<>();
List<List<Integer>> result = new ArrayList<>();
public List<List<Integer>> combine(int n, int k) {
backTrack(n,k,1);
return result;
}
void backTrack(int n, int k, int startIndex){
if(path.size() == k){
result.add(new ArrayList<>(path));
return;
}
for(int i = startIndex; i <= n; ++i){
path.addLast(i);
backTrack(n,k,i+1);
path.removeLast();
}
}
}