前端电商 sku 的全排列算法很难吗？学会这个套路，完全掌握排列组合。

前言

前段时间在掘金看到一个热帖 今天又懒得加班了，能写出这两个算法吗？带你去电商公司写商品中心，里面提到了一个比较有意思故事，大意就是一个看似比较简单的电商 sku 的全排列组合算法，可是却有好多人没能顺利写出来。有一个毕业生小伙子在面试的时候给出了思路，可是进去之后仍是没写出来，羞愧跑路~

其实排列组合是一个很经典的算法，也是对递归回溯法的一个实践运用，本篇文章就以带你学习一个标准「排列组合求解模板」，耐心看完，你会有更多收获。

需求

需求描述起来很简单，有这样三个数组：

let names = ["iPhone X", "iPhone XS"]

let colors = ["黑色", "白色"]

let storages = ["64g", "256g"]

须要把他们的全部组合穷举出来，最终获得这样一个数组：

[
  ["iPhone X", "黑色", "64g"],
  ["iPhone X", "黑色", "256g"],
  ["iPhone X", "白色", "64g"],
  ["iPhone X", "白色", "256g"],
  ["iPhone XS", "黑色", "64g"],
  ["iPhone XS", "黑色", "256g"],
  ["iPhone XS", "白色", "64g"],
  ["iPhone XS", "白色", "256g"],
]

因为这些属性数组是不定项的，因此不能简单的用三重的暴力循环来求解了。

思路

若是咱们选用递归回溯法来解决这个问题，那么最重要的问题就是设计咱们的递归函数。

思路分解

以上文所举的例子来讲，好比咱们目前的属性数组就是：names、colors、storages，首先咱们会处理 names 数组，很显然对于每一个属性数组，都须要去遍历它，而后一个一个选择后再去和 下一个数组的每一项进行组合。

咱们设计的递归函数接受两个参数：

• index 对应当前正在处理的下标，是 names 仍是 colors 或是 storage。
• prev 上一次递归已经拼接成的结果，好比 ['iPhone X', '黑色']。

进入递归函数：

1. 处理属性数组的下标0：假设咱们在第一次循环中选择了 iPhone XS，那此时咱们有一个未完成的结果状态，假设咱们叫它 prev，此时 prev = ['iPhone XS']。
2. 处理属性数组的下标1：那么就处理到 colors 数组的了，而且咱们拥有 prev，在遍历 colors 的时候继续递归的去把 prev 拼接成 prev.concat(color)，也就是 ['iPhone XS', '黑色'] 这样继续把这个 prev 交给下一次递归。
3. 处理属性数组的下标2：那么就处理到 storages 数组的了，而且咱们拥有了 name + color 的 prev，在遍历 storages 的时候继续递归的去把 prev 拼接成 prev.concat(storage)，也就是 ['iPhone XS', '黑色', '64g']，而且此时咱们发现处理的属性数组下标已经到达了末尾，那么就放入全局的结果变量 res 中，做为一个结果。

编码实现

let names = ["iPhone X", "iPhone XS"]

let colors = ["黑色", "白色"]

let storages = ["64g", "256g"]

let combine = function (...chunks) {
  let res = []

  let helper = function (chunkIndex, prev) {
    let chunk = chunks[chunkIndex]
    let isLast = chunkIndex === chunks.length - 1
    for (let val of chunk) {
      let cur = prev.concat(val)
      if (isLast) {
        // 若是已经处理到数组的最后一项了 则把拼接的结果放入返回值中
        res.push(cur)
      } else {
        helper(chunkIndex + 1, cur)
      }
    }
  }

  // 从属性数组下标为 0 开始处理
  // 而且此时的 prev 是个空数组
  helper(0, [])

  return res
}

console.log(combine(names, colors, storages))

递归树图

画出以 iPhone X 这一项为起点的递归树图，固然这个问题是一个多个根节点的树，请自行脑补 iPhone XS 为起点的树，子结构是如出一辙的。

万能模板

为何说这种接法是排列组合的「万能模板呢」？来看一下 LeetCode 上的真题。

组合-77

77. 组合 这是一道难度为 medium 的问题，其实算是比较有难度的问题了：

问题

给定两个整数 n 和 k，返回 1 ... n 中全部可能的 k 个数的组合。

示例:

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

解答

let combine = function (n, k) {
  let ret = []

  let helper = (start, prev) => {
    let len = prev.length
    if (len === k) {
      ret.push(prev)
      return
    }

    for (let i = start; i <= n; i++) {
      helper(i + 1, prev.concat(i))
    }
  }
  helper(1, [])
  return ret
}

能够看出这题和咱们求解电商排列组合的代码居然如此类似。只须要设计一个接受 start排列起始位置、prev上一次拼接结果为参数的递归 helper函数，

而后对于每个起点下标 start，先拼接上 start位置对应的值，再不断的再以其余剩余的下标做为起点去作下一次拼接。当 prev 这个中间状态的拼接数组到达题目的要求长度 k后，就放入结果数组中。

优化

在这个解法中，有一些递归分支是明显不可能获取到结果的，咱们每次递归都会循环尝试 <= n的全部项去做为start，假设咱们要求的数组长度 k = 3，最大值 n = 4。

而咱们以 prev = [1]，再去以 n = 4 为 start 做为递归的起点，那么显然是不可能获得结果的，由于 n = 4 的话就只剩下 4这一项能够拼接了，最多也就拼接成 [1, 4]，不可能知足 k = 3 的条件。

因此在进入递归以前，就果断的把这些“废枝”给减掉。这就叫作「剪枝」

let combine = function (n, k) {
  let ret = []

  let helper = (start, prev) => {
    let len = prev.length
    if (len === k) {
      ret.push(prev)
      return
    }

    // 还有 rest 个位置待填补
    let rest = k - prev.length
    for (let i = start; i <= n; i++) {
      if (n - i + 1 < rest) {
        continue
      }
      helper(i + 1, prev.concat(i))
    }
  }
  helper(1, [])
  return ret
}

类似题型

固然，力扣中能够套用这个模板的类似题型还有不少，并且大多数难度都是 medium的，好比快手的面试题子集 II-90，能够看出排列组合的递归解法仍是有必定的难度的。

我在维护的 LeetCode 题解仓库 中已经按标签筛选好 「递归与回溯」类型的几道题目和解答了，感兴趣的小伙伴也能够一块儿攻破它们。

总结

排列组合问题并非空中楼阁，在实际工做中也会常常遇到这种场景，掌握了递归回溯的标准模板固然不是为了让你死记硬背套公式，而是真正的理解它。遇到须要递归解决的问题。

1. 画出递归树状图，找出递归公式。
2. 对于不可能达成条件的分支递归，进行合理的「剪枝」。

但愿阅读完本篇文章的你，能对递归和排列组合问题有进一步的理解和收获。

❤️ 感谢你们

1.若是本文对你有帮助，就点个赞支持下吧，你的「赞」是我创做的动力。

2.关注公众号「前端从进阶到入院」便可加我好友，我拉你进「前端进阶交流群」，你们一块儿共同交流和进步。