回溯

就是暴力法都搜不出来的的递归算法

涉及到的问题

  • 组合问题:N个数里面按一定规则找出k个数的集合
  • 排列问题:N个数按一定规则全排列,有几种排列方式
  • 切割问题:一个字符串按一定规则有几种切割方式
  • 子集问题:一个N个数的集合里有多少符合条件的子集
  • 棋盘问题:N皇后,解数独等等

基本模板

for循环横向遍历,递归纵向遍历,回溯不断调整结果集

  1. function **(){
  2. const cur = []
  3. const res = []
  4. backtrack(路径)
  5. function backtrack(路径){
  6. if (终止条件) {
  7. 存放结果;
  8. return;
  9. }
  10. for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
  11. 处理节点;
  12. dfs(路径); // 递归
  13. 回溯,撤销处理结果
  14. }
  15. }
  16. return res
  17. }

优化

优化回溯算法只有剪枝一种方法

总结

1. 什么时候backtrack需要索引index?

  • 对于组合问题

    • 集合来求组合的话,就需要startIndex,例如:[回溯算法:求组合问题!]
    • 多个集合取组合,各个集合之间相互不影响,那么就不用startIndex 例如:[回溯算法:电话号码的字母组合]

2. 纵向去重

使用过的地方不能重复切割所以递归函数需要传入i + 1

  1. backtrack(i + 1);

3. 横向去重

对于可以排序的

先排序

  1. function backtrack(index) {
  2. if (sum > target) return;
  3. if (sum === target) return res.push(cur.slice());
  4. for (let i = index; i < candidates.length; i++) {
  5. // 横向去重,不会涉及纵向的重复元素
  6. if (i > index && candidates[i] === candidates[i - 1]) {
  7. continue;
  8. }
  9. const element = candidates[i];
  10. cur.push(element);
  11. sum += element;
  12. backtrack(i + 1);
  13. sum -= element;
  14. cur.pop();
  15. }
  16. }

对于可排序但是不能通过index过滤纵向的

如果要对树层中前一位去重,就用used[i - 1] == false,如果要对树枝前一位去重用used[i - 1] == true

  1. var permuteUnique = function (nums) {
  2. const cur = [];
  3. const res = [];
  4. const set = [];
  5. nums.sort((a, b) => a - b);
  6. backtrack();
  7. function backtrack() {
  8. if (nums.length === cur.length) {
  9. return res.push(cur.slice());
  10. }
  11. for (let i = 0; i < nums.length; i++) {
  12. if (set[i]) {
  13. continue;
  14. }
  15. // 去重的关键代码 !set[i - 1]
  16. if (i > 0 && nums[i] === nums[i - 1] && !set[i - 1]) {
  17. continue;
  18. }
  19. set[i] = true;
  20. cur.push(nums[i]);
  21. backtrack();
  22. set[i] = false;
  23. cur.pop();
  24. }
  25. }
  26. return res;
  27. };

对于不可以排序的

  1. // 每一次调用dfs函数 新建一个new Set()
  2. function dfs(index) {
  3. const set = new Set();
  4. if (current.length >= 2) {
  5. res.push(current.slice());
  6. }
  7. for (let i = index; i < nums.length; i++) {
  8. const element = nums[i];
  9. // 横向去重
  10. if (set.has(element)) {
  11. continue;
  12. }
  13. set.add(element);
  14. if (current.length && current[current.length - 1] > element) {
  15. continue;
  16. }
  17. current.push(element);
  18. dfs(i + 1);
  19. current.pop();
  20. }
  21. }

4. 切割问题,如何切

类似于组合问题,组合问题种,待推入的当前元素,切割问题待推入的是根据起始点,和终止点定位的一段区间

// 分割问题
const sub = s.substring(index, i + 1);
cur.push(sub);
backtrack(i + 1);
cur.pop();

// 组合问题
const sub = s[i]
cur.push(sub);
backtrack(i + 1);
cur.pop();

5. 如果涉及到数组集合有重复

就先排序

nums.sort((a, b) => a - b);

6. 排列问题是有序的,所有不用index

在排列问题种 [1,2] 和 [2,1]是不同的,需要一个哈希表存储用过的数据

// 建立哈希表
const set = new Set();
backtrack(0);
function backtrack(index) {
    if (cur.length === nums.length) {
      return res.push(cur.slice());
    }
    for (var i = 0; i < nums.length; i++) {
      if (!set.has(nums[i])) {
        // 存储当前用到的数据
        set.add(nums[i]);
        cur.push(nums[i]);
        backtrack();
        // 释放
        set.delete(nums[i]);
        cur.pop();
      }
    }
  }

组合问题

  • 时间复杂度:O(n * 2^n),组合问题其实就是一种子集的问题,所以组合问题最坏的情况,也不会超过子集问题的时间复杂度。
  • 空间复杂度:O(n),和子集问题同理。

[77] 组合

给定两个整数 nk,返回范围 [1, n] 中所有可能的 k 个数的组合。

var combine = function (n, k) {
  const cur = [];
  const res = [];
  backtrack(1);
  function backtrack(index) {
    if (cur.length === k) {
      res.push(cur.slice());
      return res;
    }
    for (let i = index; i <= n; i++) {
      cur.push(i);
      backtrack(i + 1);
      cur.pop();
    }
  }
  return res;
};

剪枝优化

var combine = function (n, k) {
  const cur = [];
  const res = [];
  backtrack(1);
  function backtrack(index) {
    if (cur.length === k) {
      res.push(cur.slice());
      return res;
    }
    for (let i = index; i <= n - (k - cur.length) + 1; i++) {
      cur.push(i);
      backtrack(i + 1);
      cur.pop();
    }
  }
  return res;
};

[216] 组合总和 III

找出所有相加之和为 nk*。*组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

说明:

  • 所有数字都是正整数。
  • 解集不能包含重复的组合。

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]

示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
var combinationSum3 = function (k, n) {
  const res = [];
  const cur = [];
  let sum = 0;
  backtrack(1);
  function backtrack(index) {
    if (sum > n) {
      return;
    }
    if (sum === n && cur.length === k) {
      res.push(cur.slice());
      return res;
    }
    for (let i = index; i <= 9 - (k - cur.length) + 1; i++) {
      cur.push(i);
      sum += i;
      backtrack(i + 1);
      sum -= i;
      cur.pop();
    }
  }
  return res;
};

[39] 组合总和

给定一个无重复元素的正整数数组 candidates 和一个正整数 target ,找出 candidates 中所有可以使数字和为目标数 target 的唯一组合。

candidates 中的数字可以无限制重复被选取。如果至少一个所选数字数量不同,则两种组合是唯一的。

对于给定的输入,保证和为 target 的唯一组合数少于 150 个。

示例 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]]

示例 3:

输入: candidates = [2], target = 1
输出: []

示例 4:

输入: candidates = [1], target = 1
输出: [[1]]

示例 5:

输入: candidates = [1], target = 2
输出: [[1,1]]
var combinationSum = function (candidates, target) {
  const cur = [];
  const res = [];
  let sum = 0;
  backtrack(0);
  function backtrack(index) {
    if (sum > target) {
      return;
    }
    if (sum === target) {
      res.push(cur.slice());
      return;
    }
    for (let i = index; i < candidates.length; i++) {
      const element = candidates[i];
      if (sum + element > target) {
        continue;
      }
      sum += element;
      cur.push(element);
      backtrack(i);
      sum -= element;
      cur.pop();
    }
  }
  return res;
};

[17] 电话号码的字母组合

给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。答案可以按 任意顺序 返回。

给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。

var letterCombinations = function (digits) {
  const map = [
    '',
    '',
    'abc',
    'def',
    'ghi',
    'jkl',
    'mno',
    'pqrs',
    'tuv',
    'wxyz',
  ];
  const cur = [];
  const res = [];
  if (!digits) return [];
  if (digits.length === 1) return map[digits].split('');
  backtrack(0);
  function backtrack(index) {
    if (cur.length === digits.length) {
      res.push(cur.slice().join(''));
      return res;
    }
    for (const item of map[digits[index]]) {
      cur.push(item);
      backtrack(index + 1);
      cur.pop();
    }
  }
  return res;
};

[40] 组合总和 II

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

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

注意:解集不能包含重复的组合。

var combinationSum2 = function (candidates, target) {
  candidates.sort((a, b) => a - b);
  const res = [];
  const cur = [];
  let sum = 0;
  backtrack(0);
  function backtrack(index) {
    if (sum > target) return;
    if (sum === target) return res.push(cur.slice());
    for (let i = index; i < candidates.length; i++) {
      if (i > index && candidates[i] === candidates[i - 1]) {
        continue;
      }
      const element = candidates[i];
      cur.push(element);
      sum += element;
      backtrack(i + 1);
      sum -= element;
      cur.pop();
    }
  }
  return res;
};

切割问题

[131].分割回文串

给你一个字符串 s,请你将 s 分割成一些子串,使每个子串都是 回文串 。返回 s 所有可能的分割方案。

回文串 是正着读和反着读都一样的字符串。

var partition = function (s) {
  const res = [];
  const cur = [];
  backtrack(0);
  function backtrack(index) {
    if (index >= s.length) {
      res.push(cur.slice());
      return;
    }
    for (let i = index; i < s.length; i++) {
      if (isPalindrome(s, index, i)) {
        const sub = s.substring(index, i + 1);
        console.log(sub);
        cur.push(sub);
        backtrack(i + 1);
        cur.pop();
      }
    }
  }
  function isPalindrome(s, l, r) {
    for (let i = l, j = r; i < j; i++, j--) {
      if (s[i] !== s[j]) {
        return false;
      }
    }
    return true;
  }
  return res;
};

[93] 复原 IP 地址

给定一个只包含数字的字符串,用以表示一个 IP 地址,返回所有可能从 s 获得的 有效 IP 地址 。你可以按任何顺序返回答案。

有效 IP 地址 正好由四个整数(每个整数位于 0 到 255 之间组成,且不能含有前导 0),整数之间用 '.' 分隔。

例如:”0.1.2.201” 和 “192.168.1.1” 是 有效 IP 地址,但是 “0.011.255.245”、”192.168.1.312” 和 “192.168@1.1” 是 无效 IP 地址。

示例 1:

输入:s = "25525511135"
输出:["255.255.11.135","255.255.111.35"]

示例 2:

输入:s = "0000"
输出:["0.0.0.0"]

示例 3:

输入:s = "1111"
输出:["1.1.1.1"]

示例 4:

输入:s = "010010"
输出:["0.10.0.10","0.100.1.0"]

示例 5:

输入:s = "101023"
输出:["1.0.10.23","1.0.102.3","10.1.0.23","10.10.2.3","101.0.2.3"]
var restoreIpAddresses = function (s) {
  const res = [];
  const cur = [];
  backtrack(0);
  function backtrack(index) {
    if (cur.length > 4) {
      return;
    }
    if (cur.length === 4 && index === s.length) {
      res.push(cur.slice().join('.'));
      return;
    }
    for (let i = index; i < s.length; i++) {
      const str = s.substring(index, i + 1);
      if (str.length > 3 || +str > 255) break;
      if (str.length > 1 && str[0] === '0') break;
      cur.push(str);
      backtrack(i + 1);
      cur.pop();
    }
  }
  return res;
};

子集问题

求取子集问题,不需要任何剪枝!因为子集就是要遍历整棵树

  • 时间复杂度:O(n 2n),构造每一组子集都需要填进数组,又有需要O(n),最终时间复杂度:O(n 2^n)
  • 空间复杂度:O(n),递归深度为n,所以系统栈所用空间为O(n),每一层递归所用的空间都是常数级别,注意代码里的result和path都是全局变量,就算是放在参数里,传的也是引用,并不会新申请内存空间,最终空间复杂度为O(n)

[78] 子集

给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。

解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。

示例 1:

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

示例 2:

输入:nums = [0]
输出:[[],[0]]
var subsets = function (nums) {
  const res = [];
  const cur = [];
  backtrack(0);
  function backtrack(index) {
    res.push(cur.slice());
    for (let i = index; i < nums.length; i++) {
      const element = nums[i];
      cur.push(element);
      backtrack(i + 1);
      cur.pop();
    }
  }
  return res;
};

[90] 子集 II

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

解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。

示例 1:

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

示例 2:

输入:nums = [0]
输出:[[],[0]]
var subsetsWithDup = function (nums) {
  const res = [];
  const cur = [];
  nums.sort((a, b) => a - b);
  backtrack(0);
  function backtrack(index) {
    res.push(cur.slice());
    for (var i = index; i < nums.length; i++) {
      if (i > index && nums[i - 1] === nums[i]) {
        continue;
      }
      cur.push(nums[i]);
      backtrack(i + 1);
      cur.pop();
    }
  }
  return res;
};

[491] 递增子序列

给定一个整型数组, 你的任务是找到所有该数组的递增子序列,递增子序列的长度至少是 2 。

示例:

输入:[4, 6, 7, 7]
输出:[[4, 6], [4, 7], [4, 6, 7], [4, 6, 7, 7], [6, 7], [6, 7, 7], [7,7], [4,7,7]]

提示:

  • 给定数组的长度不会超过15。
  • 数组中的整数范围是 [-100,100]。
  • 给定数组中可能包含重复数字,相等的数字应该被视为递增的一种情况。
var findSubsequences = function (nums) {
  const res = [];
  const current = [];
  dfs(0);

  function dfs(index) {
    const set = new Set();
    if (current.length >= 2) {
      res.push(current.slice());
    }
    for (let i = index; i < nums.length; i++) {
      const element = nums[i];
      if (set.has(element)) {
        continue;
      }
      set.add(element);
      if (current.length && current[current.length - 1] > element) {
        continue;
      }
      current.push(element);
      dfs(i + 1);
      current.pop();
    }
  }
  return res;
};

排列问题

  • 时间复杂度:O(n!),这个可以从排列的树形图中很明显发现,每一层节点为n,第二层每一个分支都延伸了n-1个分支,再往下又是n-2个分支,所以一直到叶子节点一共就是 n n-1 n-2 * ….. 1 = n!。
  • 空间复杂度:O(n),和子集问题同理。

[46] 全排列

给定一个不含重复数字的数组 nums ,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。

示例 1:

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

示例 2:

输入:nums = [0,1]
输出:[[0,1],[1,0]]

示例 3:

输入:nums = [1]
输出:[[1]]
var permute = function (nums) {
  const cur = [];
  const res = [];
  const set = new Set();
  backtrack(0);
  function backtrack(index) {
    if (cur.length === nums.length) {
      return res.push(cur.slice());
    }
    for (var i = 0; i < nums.length; i++) {
      if (!set.has(nums[i])) {
        set.add(nums[i]);
        cur.push(nums[i]);
        backtrack();
        set.delete(nums[i]);
        cur.pop();
      }
    }
  }
  return res;
};

[47] 全排列 II

给定一个可包含重复数字的序列 nums按任意顺序 返回所有不重复的全排列。

示例 1:

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

示例 2:

输入:nums = [1,2,3]
输出:[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]
var permuteUnique = function (nums) {
  const cur = [];
  const res = [];
  nums.sort((a, b) => a - b);
  let map = [];
  backtrack();

  function backtrack() {
    if (cur.length === nums.length) {
      return res.push(cur.slice());
    }
    for (let i = 0; i < nums.length; i++) {
      const curElement = nums[i];
      if (map[i]) {
        continue;
      }
      if (i > 0 && nums[i] === nums[i - 1] && !map[i - 1]) {
        continue;
      }
      map[i] = true;
      cur.push(curElement);
      backtrack();
      map[i] = false;
      cur.pop();
    }
  }
  return res;
};

棋盘问题

重新安排行程

给你一份航线列表 tickets ,其中 tickets[i] = [fromi, toi] 表示飞机出发和降落的机场地点。请你对该行程进行重新规划排序。

所有这些机票都属于一个从 JFK(肯尼迪国际机场)出发的先生,所以该行程必须从 JFK 开始。如果存在多种有效的行程,请你按字典排序返回最小的行程组合。

  • 例如,行程 ["JFK", "LGA"]["JFK", "LGB"] 相比就更小,排序更靠前。

假定所有机票至少存在一种合理的行程。且所有的机票 必须都用一次 且 只能用一次。

输入:tickets = [["MUC","LHR"],["JFK","MUC"],["SFO","SJC"],["LHR","SFO"]]
输出:["JFK","MUC","LHR","SFO","SJC"]
输入:tickets = [["JFK","SFO"],["JFK","ATL"],["SFO","ATL"],["ATL","JFK"],["ATL","SFO"]]
输出:["JFK","ATL","JFK","SFO","ATL","SFO"]
解释:另一种有效的行程是 ["JFK","SFO","ATL","JFK","ATL","SFO"] ,但是它字典排序更大更靠后。
var findItinerary = function (tickets) {
  const result = ['JFK'];
  const map = {};
  for (const [from, to] of tickets) {
    if (!map[from]) {
      map[from] = [];
    }
    map[from].push(to);
  }
  Object.keys(map).forEach((from) => {
    map[from].sort();
  });
  backtrack();
  function backtrack() {
    if (result.length === tickets.length + 1) {
      return true;
    }
    if (
      !map[result[result.length - 1]] ||
      !map[result[result.length - 1]].length
    ) {
      return false;
    }
    for (let i = 0; i < map[result[result.length - 1]].length; i++) {
      let city = map[result[result.length - 1]][i];
      // 删除已走过航线,防止死循环
      map[result[result.length - 1]].splice(i, 1);
      result.push(city);
      if (backtrack()) {
        return true;
      }
      result.pop();
      map[result[result.length - 1]].splice(i, 0, city);
    }
  }
  return result;
};

[51] N 皇后

Tagsbacktracking

CompaniesUnknown

n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。

给你一个整数 n ,返回所有不同的 n 皇后问题 的解决方案。

每一种解法包含一个不同的 n 皇后问题 的棋子放置方案,该方案中 'Q''.' 分别代表了皇后和空位。

var solveNQueens = function (n) {
  function isValid(row, col, chessBoard, n) {
    // 检查列
    for (let i = 0; i < row; i++) {
      if (chessBoard[i][col] === 'Q') {
        return false;
      }
    }
    // 检查 45度角是否有皇后
    for (let i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
      if (chessBoard[i][j] === 'Q') {
        return false;
      }
    }
    // 检查 135度角是否有皇后
    for (let i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
      if (chessBoard[i][j] === 'Q') {
        return false;
      }
    }
    return true;
  }
  function transformChessBoard(chessBoard) {
    return chessBoard.map((item) => item.join(''));
  }
  const cur = new Array(n).fill([]).map(() => new Array(n).fill('.'));
  const res = [];
  dfs(0);
  function dfs(row) {
    if (row == n) {
      res.push(transformChessBoard(cur));
      return;
    }
    for (let col = 0; col < n; col++) {
      if (isValid(row, col, cur, n)) {
        cur[row][col] = 'Q';
        dfs(row + 1);
        cur[row][col] = '.';
      }
    }
  }
  return res;
};