小白聊回溯

春困秋乏夏打盹,现在正是会经常打盹儿的时候,为了换换脑子振奋一下精神,默默打开了leetcode练练脑子。

一道组合总和瞪着大眼看了半天,用递归试了又试结果就是不对,无奈还是去Google一下吧,那能怎么办呢,谁让咱是小菜鸟呢

一看才知道这是用回溯解决啊,先看看维基百科对回溯法的解释吧:

回溯法是暴力搜索法的一种

对于某些计算问题而言,回溯法是一种可以找出所有(或一部分)解的一般性算法,尤其适用于约束满足问题(在解决约束满足问题时,我们逐步构造更多的候选解,并且在确定某一部分候选解不可能补全成正确解之后放弃继续搜索这个部分候选解本身及其可以拓展出的子候选解,转而测试其他的部分候选解)

回溯法采用试错的思想,它尝试分步的去解决一个问题。在分步解决问题的过程中,当它通过尝试发现现有的分步答案不能得到有效的正确解答的时候,它将取消上一步甚至上几步的计算,再通过其他的可能的分步解答再次尝试寻找问题的答案。回溯法常用最简单的递归方法来实现,在反复重复上述的步骤后可能出现两种情况:

- 找到一个可能存在的正确答案
- 在尝试了所有可能的分步方法后宣告该问题没有答案

小白理解:回溯法就是按照深度优先搜索,按一条路一直往下走,遇到正确解了就加入结果集,然后后退一步或几步继续查找其他解;如果发现不满足条件了,就后退一步或几步选另一条路往下走。

光看文字解释也整不太明白(反正我是这样的。。),实践才是真道理,看一下相关的题目吧。

1. 组合总和

问题描述:给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的数字可以无限制重复被选取。

说明:

所有数字(包括 target)都是正整数。
解集不能包含重复的组合。 
var combinationSum = function(candidates, target) {
    var res = [];
    candidates = candidates.sort((a, b) => a-b);
    find(candidates, target, res, [], 0)
    return res;
}
/**
* arr: 数组
* target:目标数
* res: 结果集
* path: 候选解
* j: for循环的起始下标
*/
function find (arr, target, res, path, j) {
    var sum = getSum(path);
    if (sum === target) { // 符合条件,加入结果集
        res.push([...path]);
        return;
    }
    if (sum > target) { // 大于目标值,直接返回
        return;
    }
    // 下标从j开始,避免解集重复。比如arr是[2,3,6,7],target是7,开始得到一个正解是[2,2,3],当候选解第一个元素是3时,如果i还从0开始,就会得到[3,2,2], 使解集重复
    for (var i = j; i < arr.length; i++) {
        path.push(arr[i]);
        find(arr, target, res, path, i);
        // 无论是该路径大于target还是等于target,都需要对其删除最后一个元素,进行其余支路的搜索
        path.pop();
    }
}
function getSum(path) {
    var sum = path.reduce((s, n) => {
        return s += n;
    }, 0)
    return sum;
}

分析:

  1. 定义全局结果数组:var res = []
  2. 调用递归函数:find
  3. 返回结果集:res
  4. 定义递归函数
    • 参数
    • 终止条件:sum === target,把正解加入结果集中
    • 剪枝条件:sum > target,过滤掉不正确的
    • 回溯搜索下一结果:path.pop(),删除最后一个,进行其余支路的搜索

2. 组合总和II

给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用一次。

说明:

所有数字(包括目标数)都是正整数。
解集不能包含重复的组合。 

分析

这道题跟上题的区别在于:

  1. 数组 candidates 里面的元素可以重复
  2. 数组中的每个元素在每个组合中只能使用一次

因为数组元素可以重复,就可能导致解集重复。比如candidates = [10,1,2,7,6,1,5],target = 8;其中正解包括[1,7], [7,1], 这两个就是重复的,第一个解的1是candidates中的第二个元素,第二个解的1是candidates中的第6个元素。

针对这个问题,我们可以先对数组排序,这样相同元素位置紧邻。这样上面的两个正解就变成:[1,7],[1,7];我们根据下面这个if条件:

if (i > j && arr[i] === arr[i - 1]) {
    continue
} 

过滤掉第二个相同的解。

针对第二个问题:每个元素在每个组合中只能使用一次。我们在进行递归的时候最后一个参数的传参为i + 1, 这样就解决了这个问题。

代码:

var combinationSum2 = function(candidates, target) {
   var res = [];
   candidates = candidates.sort((a,b) => a-b)
   find(candidates, target, res, [], 0)
   return res;
};

function find(arr,target, res, path, j) {
    var sum = getSum(path);
    if (sum === target) {
        res.push([...path]);
        return;
    }
    if (sum > target) {
        return;
    }
    for (var i = j; i < arr.length; i++) {
        if (i > j && arr[i] === arr[i - 1]) { // 保证解集不重复
            continue
        } 
        path.push(arr[i]);
        find(arr,target, res, path, i + 1); // i + 1保证 每个数字在每个组合中只能使用一次。
        path.pop();
    }
}
function getSum (path) {
    var sum = path.reduce((s, num) => {
        s += num
        return s
    }, 0)
    return sum;
}

3. 全排列

题目描述:

给定一个没有重复数字的序列,返回其所有可能的全排列。

示例:

输入: [1,2,3]
输出:
[
  [1,2,3],
  [1,3,2],
  [2,1,3],
  [2,3,1],
  [3,1,2],
  [3,2,1]
]

分析:

首先递归的终止条件是:path.length === nums.length,如果符合条件就加入结果集当中。

为了保证每个元素在候选解中只出现一次,我们用一个visited对象保存状态,记录某个元素是否使用过。

var permute = function(nums) {
    const res = []
    const path = []
    range (nums, res, path, {})
    return res;
 };
function range (nums, res, path, visited) {
    if (path.length === nums.length) {
        res.push([...path])
        return;
    }

    for(var i = 0; i < nums.length; i++) {
        if (!visited[i]) {
            visited[i] = true
            path.push(nums[i])
            range(nums, res, path, visited)
            path.pop()
            visited[i] = false
        }
    }
}

4. 全排列II

题目描述:

给定一个可包含重复数字的序列,返回所有不重复的全排列。

示例:

输入: [1,1,2]
输出:
[
  [1,1,2],
  [1,2,1],
  [2,1,1]
]

分析:

该题跟上一题的不同就是数组可包含重复数字,所以要在上一题的基础上去除重复排列。

首先把数组排序,然后添加如下判断:

 if (visited[i] || (i > 0 && nums[i] === nums[i - 1] &&!visited[i - 1])) continue

在visited[i]的基础上添加了(i > 0 && nums[i] === nums[i - 1] &&!visited[i - 1])

如果不加这个条件,当数组是[1,1,2]时,正解包含两个[1,1,2], 一个下标是0 1 2的排列,一个下标是1 0 2的排列,由于数组前两个元素都是1,两种排列会导致一样的结果。我们加上面那个条件的作用是只保留下标顺序排列的一个解。

var permuteUnique = function(nums) {
    const res = []
    const path = []
    nums.sort((a,b) => a-b)
    range (nums, res, path, {})
    return res;
};
function range (nums, res, path, visited) {
    if (path.length === nums.length) {
        res.push([...path])
        return;
    }

    for(var i = 0; i < nums.length; i++) {
        if (visited[i] || (i > 0 && nums[i] === nums[i - 1] &&!visited[i - 1])) continue

        visited[i] = true
        path.push(nums[i])
        range(nums, res, path, visited)
        path.pop()
        visited[i] = false
    }
}

5. 子集

题目描述:

给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。

说明:解集不能包含重复的子集。

示例:

输入: nums = [1,2,3]
输出:
[
  [3],
  [1],
  [2],
  [1,2,3],
  [1,3],
  [2,3],
  [1,2],
  []
]

这是求子集,所以每次递归的时候都把path加进结果集

var subsets = function(nums) {
    var res = [];
    var path = [];
    find(nums, res, path, 0)
    return res;
};
function find (nums, res, path, j) {
    res.push([...path])

    for (var i = j; i < nums.length; i++) {
        path.push(nums[i])
        find(nums, res, path, i + 1)
        path.pop();
    }
}

6. 子集II

题目描述:

给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。

说明:解集不能包含重复的子集。

示例:

输入: [1,2,2]
输出:
[
  [2],
  [1],
  [1,2,2],
  [2,2],
  [1,2],
  []
]

分析:

该题比上一题多了一个条件:数组可能包含重复元素。

我们根据之前的经验,添加一个判断条件:

if (i > j && nums[i] === nums[i-1]) continue

只保留顺序添加的一个正解。

var subsetsWithDup = function(nums) {
    var res = [];
    var path = [];
    nums.sort((a,b) => a-b)
    find(nums, res, path, 0)
    return res;
};
function find (nums, res, path, j) {
    res.push([...path])

    for (var i = j; i < nums.length; i++) {
        if (i > j && nums[i] === nums[i-1]) continue
        path.push(nums[i])
        find(nums, res, path, i + 1)
        path.pop();
    }
}

经过上面几道题,回溯法的解题步骤一般是:

  1. 定义全局结果数组
  2. 调用递归函数
  3. 返回结果集
  4. 定义递归函数
    • 确定参数
    • 确定终止条件
    • 剪枝条件
    • 回溯搜索下一结果:path.pop(),删除最后一个,进行其余支路的搜索

当数组包含重复元素时,可以利用类似if (i > j && nums[i] === nums[i-1]) continue这种条件判断来去重

如有不恰当的地方,欢迎指正~~

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,222评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,455评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,720评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,568评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,696评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,879评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,028评论 3 409
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,773评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,220评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,550评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,697评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,360评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,002评论 3 315
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,782评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,010评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,433评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,587评论 2 350

推荐阅读更多精彩内容