给定两个整数 n 和 k,返回 1 … n 中所有可能的 k 个数的组合。
题解:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
这道题与leetcode17(求电话号码的字符组合)实际上解题思路是一致的,其实也可以这么说,在涉及枚举+组合的问题中,回溯法的普适性是很强的。
题解:维护一个栈,保存组合情况。利用回溯法(递归),穷举所有组合情况,当栈中的元素个数为k时,将栈中的结果插入结果集。
class Solution {
private final List<List<Integer>>res=new ArrayList<>();
public List<List<Integer>> combine(int n, int k) {
DFS(new boolean[n],new Stack<>(),n,k,1);
return res;
}
private void DFS(boolean[]search,Stack<Integer>subRes,int n,int k,int min){
if(subRes.size()+n-min+1<k)
return;
if(subRes.size()==k) {
res.add(new ArrayList<>(subRes));
return;
}
for(int i=min;i<=n;i++){
if(!search[i-1]) {
subRes.add(i);
search[i-1]=true;
DFS(search,subRes,n,k,i+1);
search[i-1]=false;
subRes.pop();
}
}
}
}