前面二叉树还没步入正轨,为了尽量赶上进度,觉得先进行下一章节的学习,周末时间补回之前落下的二叉树的进度。
今天开始的回溯算法,我在之前也有过接触,但没有完整地写出过代码。借着这次进度,我也接着复习一下。
回溯算法基本步骤
回溯三部曲:
- 确定递归函数参数
- 确定终止条件
- 确定单层递归逻辑
解决问题:组合、切割、子集、排列集合、棋盘
算法模板(看成树):
Void backtracking(参数){
If(终止条件){
收集结果;
Return;}
For(集合元素){
处理节点;
递归函数;
回溯操作;(撤销本次处理节点情况)}
Return}
开始做题!
77.组合
给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]
思路:碰到这样的问题,我们可以利用树的结构进行画图,尽可能例举出所有可能。
首先定义result与path,记录结果集和每一步获取的集合。
代码实现:
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
class Solution {
List<List<Integer>> result= new ArrayList<>();
LinkedList<Integer>path=new LinkedList<>();
public List<List<Integer>> combine(int n, int k) {
backtracking(n,k,1);
return result;
}
public void backtracking(int n, int k, int startIndex){
if(path.size()==k){
result.add(new ArrayList<>(path));//将LinkedList的path转化为Arraylist类型,与result匹配
return;
}
for (int i = startIndex; i <=n; i++) {
path.add(i);
backtracking(n,k,i+1);//不可简化写成i++
path.removeLast();
}
}
}
总结:要熟悉回溯算法的基本步骤,以及各种集合的相互转化。