大家好我西瓜哥,今天做一道比较常考的算法题。
编写一种方法,计算某字符串的所有排列组合,字符串每个字符均不相同。
如输入 "qwe"
,要求返回 ["qwe", "qew", "wqe", "weq", "ewq", "eqw"]
。
LeetCode 题目链接:
https://leetcode-cn.com/problems/permutation-i-lcci/
回溯解法
常见的解法是回溯。
我们使用一个 toVisited 数组,用于维护递归过程中尚未访问的字符。
回溯函数的 TypeScript 签名如下:
(combs: string[], s: string, toVisit: string[]) => void
- combs:字符串数组。排列组合被保存的数组,回溯结束后用于返回
- s:字符串
- toVisit:尚未访问的字符数组。
先看下回溯函数的代码实现。
function r(combs, s, toVisit) { if (toVisit.length === 0) { combs.push(s); return; } for (let i = 0; i < toVisit.length; i++) { r(combs, s + toVisit[i], toVisit.filter(item => item != toVisit[i])); } }
我们会遍历 toVisit 数组,将这些尚未被访问的字符,追加到 s 末尾,然后生成一个新的移除了该字符的 toVisited,作为下一个递归调用的参数。
当所有字符都被访问过了(toVisite 数组为空)时,我们就拿到了一种排列方式,将它放到 combs,然后结束当前的这一个迭代。
当所有的迭代都结束后,我们的 combs 中就包含了所有的排列,将其返回即可。
完整代码实现如下:
function permutation(S: string): string[] { const combs = []; r(combs, '', [...S]); return combs; }; function r(combs: string[], s: string, toVisit: string[]) { if (toVisit.length === 0) { combs.push(s); return; } for (let i = 0; i < toVisit.length; i++) { r(combs, s + toVisit[i], toVisit.filter(item => item != toVisit[i])); } }
我们注意到,每次递归,都要拷贝一份待访问字符数组,即 toVisit.filter(item => item != toVisit[i])
,这个效率实在是不怎么高啊。
其实这个可以优化一下。如果你实现过排序算法,比如选择排序,你就会想到一个原地使用原数组的思路。排序算法中将原数组的元素进行交换,分为 “排序区间” 和 “未排序区间”。
我们这里也可以致敬一下,也交换一下嘛,在原数组中原地分为 “已访问字符区间” 和 “待访问数组”。然后在下一个递归函数执行完后,再复原一下数组。
改良后的实现如下:
function permutation(S): string[] { const combs = []; r(combs, [...S], 0); return combs; }; function r(combs, chars, first) { if (first === chars.length) { combs.push(chars.join('')); return; } for (let i = first; i < chars.length; i++) { swap(chars, i, first); r(combs, chars, first + 1); swap(chars, i, first); } } function swap(chars, i, j) { const tmp = chars[i]; chars[i] = chars[j]; chars[j] = tmp; }
结尾
本文讲解了如何用回溯的思考解全排列问题,回溯解法的核心思路是每次迭代,从待访问字符集中挑选字符,直到所有字符都被访问完。
为此你需要维护待访问字符集数组,这里如果想要高效一些,可以使用数组原地分区或是使用栈的方式,直接拷贝一份效率很低,但优点是简单直接(所以我做题时经常这么干)
我是每周做五道算法题的前端西瓜哥,欢迎关注我。