前端仔一只,不时刷刷算法题防止老年痴呆。本文是个人算法系列中的一篇,如果想了解更多关于算法的内容,请点击博主的算法专栏查看。
注意:本文所有的代码都采用 JavaScript。所有题目来自 leetcode,全部答案通过了测试,可放心食用。
理论回顾
回溯法(Backtrack)其实是基于递归来实现的。
但是它的思考逻辑很有意思,和走迷宫一样。比如我们走到一个分叉口,我们不知道哪一个路口是正确的,但是我们可以先随便选择一个路口。如果最后走不通,我们可以原地返回,在之前的分叉口重新抉择。
正是因为这种回溯的思考方式,所以这种算法称之为回溯法。
笼统地讲,回溯算法很多时候都应用在“搜索”这类问题上。不过这里说的搜索,并不是狭义的指我们前面讲过的图的搜索算法,而是在一组可能的解中,搜索满足期望的解。
真题演练
组合总和
题目
给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的数字可以无限制重复被选取。
说明:
- 所有数字(包括 target)都是正整数。
- 解集不能包含重复的组合。
示例 1:
输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
[7],
[2,2,3]
]
示例 2:
输入: candidates = [2,3,5], target = 8,
所求解集为:
[
[2,2,2,2],
[2,3,3],
[3,5]
]
解答
这一题的解题思路很简单,我们直接暴力枚举即可。其实我们开头说到了,枚举是回溯法的体现,回溯法的关键之处在于“回”,就是什么时候条件终止,回头在找。话不多说,直接上代码。
/**
* @param {number[]} candidates
* @param {number} target
* @return {number[][]}
*/
var combinationSum = function(candidates, target) {
const ans = [], list = []
backtrack(candidates, target, 0, list, ans)
return ans
};
function backtrack(candidates, target, start, list, ans) {
const total = list.reduce((total, a) => {
total += a
return total
}, 0)
if (total >= target) {
if (total === target) {
// 因为是引用类型,做一次浅拷贝操作
ans.push(list.slice(0))
}
// 这里开始回溯
return
}
const n = candidates.