组合问题
给定两个整数 n 和 k,返回 1 … n 中所有可能的 k 个数的组合。
求解过程
首先我们不难想到使用回溯法来实现该组合过程,并通过某种限制条件以节约查找次数,例如在10个数字中选择5个数字时,如果第二个数字已经查找到8,也就是说剩下的数字就算是全部添加也不能将5个数字填满(1,8,9,10),直接返回上一级。
即如果不满足k - ans.size() < n - index + 1
的情况可以直接返回。
class Solution {
List<List<Integer>> all = new LinkedList<>();
List<Integer> ans = new LinkedList<>();
public List<List<Integer>> combine(int n, int k) {
findAns(n, k, 1);
return all;
}
public void findAns(int n, int k, int index){
//约束条件:如果在这个数字之后剩下的数组不能填齐剩下的k-ans.size()个位置,则可以直接结束该查找,以节省查找次数
if(k - ans.size() > n - index + 1){
return;
}
//结束条件:当链表长度等于k时,将该结果放入到最终的答案列表中
if(ans.size()==k){
all.add(new LinkedList(ans));
return;
}
//使用递归的方式向链表传值,每次传入一个值并递归传入第二个值,依次类推,当完成最后一级查找时,清除上一级查找的结果并继续该位置元素的查找。
for(int i = index; i<=n; i++){
ans.add(i);
findAns(n, k, i+1);
ans.remove(ans.size()-1);
}
}
}