题目0046:全排列

题目描述

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

示例:

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

解题技巧

回溯法:一种通过探索所有可能的候选解来找出所有的解的算法。如果候选解被确认不是一个解的话(或者至少不是最后一个解),回溯算法会通过在上一步进行一些变化抛弃该解,即回溯并且再次尝试。

思路和算法:这个问题可以看作有n个排列成一行的空格,我们需要从左往右依此填入题目给定的n个数,每个数只能使用一次。那么很直接的可以想到一种穷举的算法,即从左往右每一个位置都依此尝试填入一个数,看能不能填完这n个空格,在程序中我们可以用回溯法来模拟这个过程。

我们定义递归函数backtrack(first, output)表示从左往右填到第first个位置,当前排列为output。那么整个递归函数分为两个情况:

如果first==n,说明我们已经填完了n个位置(注意下标从0开始),找到了一个可行的解,我们将output放入答案数组中,递归结束。

如果first< n,我们要考虑这第first个位置我们要填哪个数。根据题目要求我们肯定不能填已经填过的数,因此很容易想到的一个处理手段是我们定义一个标记数组vis[]来标记已经填过的数,那么在填第first个数的时候我们遍历题目给定的n个数,如果这个数没有被标记过,我们就尝试填入,并将其标记,继续尝试填下一个位置,即调用函数backtrack(first + 1, output)。搜索回溯的时候要撤销这一个位置填的数以及标记,并继续尝试其他没被标记过的数。

使用标记数组来处理填过的数是一个很直观的思路,但是可不可以去掉这个标记数组呢?毕竟标记数组也增加了我们算法的空间复杂度。答案是可以的,我们可以将题目给定的n个数的数组nums[]划分成左右两个部分,左边的表示已经填过的数,右边表示待填的数,我们在递归搜索的时候只要动态维护这个数组即可。

具体来说,假设我们已经填到第first个位置.那么nums[]数组中[0,first−1]是已填过的数的集合,[first,n−1]是待填的数的集合。我们肯定是尝试用[first,n−1]里的数去填第first个数,假设待填的数的下标为i,那么填完以后我们将第i个数和第first个数交换,即能使得在填第first+1个数的时候nums[]数组的[0,first]部分为已填过的数,[first+1,n−1]为待填的数,回溯的时候交换回来即能完成撤销操作。

举个简单的例子,假设我们有[2, 5, 8, 9, 10]这5个数要填入,已经填到第3个位置,已经填了[8,9]两个数,那么这个数组目前为[8, 9 | 2, 5, 10]这样的状态,分隔符区分了左右两个部分。假设这个位置我们要填10这个数,为了维护数组,我们将2和10交换,即能使得数组继续保持分隔符左边的数已经填过,右边的待填[8, 9, 10 | 2, 5]。

当然善于思考的读者肯定已经发现这样生成的全排列并不是按字典序存储在答案数组中的,如果题目要求按字典序输出,那么请还是用标记数组或者其他方法。

下面的图展示了搜索的整个过程:

class Solution:
    def permute(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        def backtrack(first = 0):
            # 所有数都填完了
            if first == n:  
                res.append(nums[:])
            for i in range(first, n):
                # 动态维护数组
                nums[first], nums[i] = nums[i], nums[first]
                # 继续递归填下一个数
                backtrack(first + 1)
                # 撤销操作
                nums[first], nums[i] = nums[i], nums[first]

        n = len(nums)
        res = []
        backtrack()
        return res

复杂度分析

时间复杂度:O(n * n!),其中n为序列的长度。算法的复杂度首先受backtrack的调用次数制约,backtrack的调用次数为\sum_{k = 1}^{n}{P(n, k)}次,其中P(n, k) = \frac{n!}{(n - k)!} = n (n - 1) ... (n - k + 1),该式被称作n的k-排列,或者部分排列。

\sum_{k = 1}^{n}{P(n, k)} = n! + \frac{n!}{1!} + \frac{n!}{2!} + \frac{n!}{3!} + ... + \frac{n!}{(n-1)!} < 2n! + \frac{n!}{2} + \frac{n!}{2^2} + \frac{n!}{2^{n-2}} < 3n!

这说明backtrack的调用次数是O(n!)的。而对于backtrack调用的每个叶结点(共n!个),我们需要将当前答案使用O(n)的时间复制到答案数组中,相乘得时间复杂度为O(n * n!)。因此时间复杂度为 O(n * n!)O(n∗n!)。

空间复杂度:O(n),其中n为序列的长度。除答案数组以外,递归函数在递归过程中需要为每一层递归函数分配栈空间,所以这里需要额外的空间且该空间取决于递归的深度,这里可知递归调用深度为O(n)。