77.组合
给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2
输出:
[[2,4], [3,4], [2,3], [1,2], [1,3], [1,4]]
这道题是典型的需要用到回溯的题,一说到回溯就会牵扯到递归。关于递归方法的写法,我这里总结了一个套路,这个套路的原型是Carl在《代码随想录》里面的递归三部曲,不过我个人给它加了一步,算是更方便我理解吧。在这里我分享给大家:
递归四步:
一、明确递归方法的作用
二、确定参数和返回值
三、确定终止条件
四、确定单层逻辑
class Solution {
List<List<Integer>> result = null;//存放所有组合
List<Integer> path = null;//存放单个组合
public void combine1(int n, int k, int startIndex){
//递归作用:在[startIndex,n]区间中寻找k个数的组合,注意这里的起始位置是startIndex。
//参数:n,k意义同题意
//终止条件:path的长度到达k说明组合中的元素的数量已经到达了k,即找到了一种组合,就把这个组合放到result里。
//单层逻辑:取这一层未操作时的状态为初始状态,判断是否终止,不终止从startIndex开始取数,取完一个数后再对下一个位置进行取数,之后再回溯。
if(path.size() == k){//path的长度到达k说明组合中的元素的数量已经到达了k,即找到了一种组合。
result.add(new ArrayList(path));//把这个组合放到result里
return;
}
for(int i = startIndex; i <= (n - (k - path.size())) + 1; i++){//从startIndex开始取数,取到剩下的元素的数量(n+1-i)小于这个组合还需要取的数量(k - path.size()),即n+1-i<k - path.size()整理后就是上面的for中的判断句了,就不用取了,因为长度不够
path.add(i);//在这一层取数
combine1(n, k, i + 1);//这一层取完,再到[i + 1, n]这个区间继续取,一直取到长度够了就终止
path.remove(path.size() - 1);//回溯,回到这一层的初始状态,即path要把上面add的i去掉。操作完后回到初始状态是回溯算法的一个最大的特点
}
}
public List<List<Integer>> combine(int n, int k) {
result = new ArrayList<>();
path = new ArrayList<>();
combine1(n, k, 1);//方法的作用是从[1,n]这个区间中取k个数的组合,从1开始取
return result;
}
}
个人认为回溯算法最大的特点就是操作后要将方法中的参数(这里是path)回到初始的状态,这一点算是刷题的感悟吧。