(一)题目描述
给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
你可以按 任何顺序 返回答案。
(二)思路分析
该题是一个组合问题,重点概括如下:
1)如果解决一个问题有多个步骤,每一个步骤有多种方法,题目又要我们找出所有的方法,可以使用回溯算法;
2)回溯算法是在一棵树上的 深度优先遍历(因为要找所有的解,所以需要遍历);
3)组合问题,相对于排列问题而言,不计较一个组合内元素的顺序性(即 [1, 2, 3] 与 [1, 3, 2] 认为是同一个组合),因此很多时候需要按某种顺序展开搜索,这样才能做到不重不漏。
回溯算法首先需要画出递归树,不同的树决定了不同的代码实现。该题的递归树如下:
(三)代码如下(其中有详细注释):
class Solution {
public List<List<Integer>> combine(int n, int k) {
// (1)若k<=0或取出的数的个数k大于数组总个数n,这样的组合数不存在
List<List<Integer>> res=new ArrayList<>();
if (k<=0||k>n){
return res;
}
List<Integer> path=new ArrayList<>();
dfs(res,k,n,1,path);
return res;
}
//depth:当前数组里面装了几个数
//begin:从什么地方开始搜
//path为当前路径
public void dfs(List<List<Integer>> res,int k,int n,int begin,List<Integer> path){
//装满即退出当前一层的递归
if(path.size()==k){
res.add(new ArrayList<>(path));
return;
}
for (int i = begin; i <=n-(k-path.size())+1 ; i++) {
path.add(i);
dfs(res,k,n,i+1,path);
path.remove(path.size()-1);
}
}
}
参考:力扣