题目主要信息:
  • 给定一个数组,求这组数字的全排列
  • 数组无重复元素
  • 以数字在数组中的位置靠前为优先级,按字典序排列输出
举一反三:

学习完本题的思路你可以解决如下题目:

BM56. 有重复项数字的全排列

BM58. 字符串的排列

BM60. 括号生成

递归+回溯(推荐使用)

知识点:递归与回溯

递归是一个过程或函数在其定义或说明中有直接或间接调用自身的一种方法,它通常把一个大型复杂的问题层层转化为一个与原问题相似的规模较小的问题来求解。因此递归过程,最重要的就是查看能不能讲原本的问题分解为更小的子问题,这是使用递归的关键。

如果是线型递归,子问题直接回到父问题不需要回溯,但是如果是树型递归,父问题有很多分支,我需要从子问题回到父问题,进入另一个子问题。因此回溯是指在递归过程中,从某一分支的子问题回到父问题进入父问题的另一子问题分支,因为有时候进入第一个子问题的时候修改过一些变量,因此回溯的时候会要求改回父问题时的样子才能进入第二子问题分支。

思路:

全排列就是对数组元素交换位置,使每一种排列都可能出现。因为题目要求按照字典序排列输出,那毫无疑问第一个排列就是数组的升序排列,它的字典序最小,后续每个元素与它后面的元素交换一次位置就是一种排列情况,但是如果要保持原来的位置不变,那就不应该从它后面的元素开始交换而是从自己开始交换才能保证原来的位置不变,不会漏情况。

如何保证每个元素能和从自己开始后的每个元素都交换位置,这种时候我们可以考虑递归。为什么可以使用递归?我们可以看数组[1,2,3,4],如果遍历经过一个元素2以后,那就相当于我们确定了数组到该元素为止的前半部分,前半部分1和2的位置都不用变了,只需要对3,4进行排列,这对于后半部分而言同样是一个全排列,同样要对从每个元素开始往后交换位置,因此后面部分就是一个子问题。那我们考虑递归的几个条件:

  • 终止条件: 要交换位置的下标到了数组末尾,没有可交换的了,那这就构成了一个排列情况,可以加入输出数组。
  • 返回值: 每一级的子问题应该把什么东西传递给父问题呢,这个题中我们是交换数组元素位置,前面已经确定好位置的元素就是我们返还给父问题的结果,后续递归下去会逐渐把整个数组位置都确定,形成一种排列情况。
  • 本级任务: 每一级需要做的就是遍历从它开始的后续元素,每一级就与它交换一次位置。

如果只是使用递归,我们会发现,上例中的1与3交换位置以后,如果2再与4交换位置的时候,我们只能得到3412这种排列,无法得到1432这种情况。这是因为遍历的时候1与3交换位置在2与4交换位置前面,递归过程中就默认将后者看成了前者的子问题,但是其实我们1与3没有交换位置的情况都没有结束,相当于上述图示中只进行了第一个分支。因此我们用到了回溯。处理完1与3交换位置的子问题以后,我们再将其交换回原来的情况,相当于上述图示回到了父节点,那后续完整的情况交换我们也能得到。

//遍历后续的元素
for(int i = index; i < num.size(); i++){ 
    //交换二者
    swap(num, i, index); 
    //继续往后找
    recursion(res, num, index + 1); 
    //回溯
    swap(num, i, index); 
}

具体做法:

  • step 1:先将数组排序,获取字典序最小的排列情况。
  • step 2:递归的时候根据当前下标,遍历后续的元素,交换二者位置后,进入下一层递归。
  • step 3:处理完一分支的递归后,将交换的情况再换回来进行回溯,进入其他分支。
  • step 4:当前下标到达数组末尾就是一种排列情况。

图示:

alt

Java实现代码:

import java.util.*;
public class Solution {
    //交换位置函数
    private void swap(ArrayList<Integer> num, int i1, int i2{ 
        int temp = num.get(i1);
        num.set(i1, num.get(i2));
        num.set(i2, temp);
    }
    
    public void recursion(ArrayList<ArrayList<Integer>> res, ArrayList<Integer> num, int index){
        //分枝进入结尾,找到一种排列
        if(index == num.size() - 1){
            res.add(num);
        }
        else{
            //遍历后续的元素
            for(int i = index; i < num.size(); i++){ 
                //交换二者
                swap(num, i, index); 
                //继续往后找
                recursion(res, num, index + 1); 
                //回溯
                swap(num, i, index); 
            }
        }
    }
    
    public ArrayList<ArrayList<Integer>> permute(int[] num) {
        //先按字典序排序
        Arrays.sort(num);  
        ArrayList<ArrayList<Integer> > res = new ArrayList<ArrayList<Integer>>();
        ArrayList<Integer> nums = new ArrayList<Integer>();
        //数组转ArrayList
        for(int i = 0; i < num.length; i++) 
            nums.add(num[i]);
        recursion(res, nums, 0);
        return res;
    }
}

C++实现代码:

class Solution {
public:
    void recursion(vector<vector<int> > &res, vector<int> &num, int index){
        //分枝进入结尾,找到一种排列
        if(index == num.size() - 1) 
            res.push_back(num);
        else{
            //遍历后续的元素
            for(int i = index; i < num.size(); i++){ 
                //交换二者
                swap(num[i], num[index]); 
                //继续往后找
                recursion(res, num, index + 1); 
                //回溯
                swap(num[i], num[index]); 
            }
        }
    }
    
    vector<vector<int> > permute(vector<int> &num) {
        //先按字典序排序
        sort(num.begin(), num.end()); 
        vector<vector<int> > res;
        //递归获取
        recursion(res, num, 0); 
        return res;
    }
};

Python实现代码:

class Solution:
    def recursion(self, res:List[List[int]], num:List[int], index:int):
        #分枝进入结尾,找到一种排列
        if index == len(num) - 1:  
            res.append(num)
        else:
            #遍历后续的元素
            for i in range(index, len(num)): 
                #交换二者
                temp = num[i]
                num[i] = num[index]
                num[index] = temp
                #继续往后找
                self.recursion(res, num, index + 1) 
                #回溯
                temp = num[i]
                num[i] = num[index]
                num[index] = temp

    def permute(self , num: List[int]) -> List[List[int]]:
        #先按字典序排序
        num.sort() 
        res = list(list())
        #递归获取
        self.recursion(res, num, 0) 
        return res

复杂度分析:

  • 时间复杂度:O(nn!)O(n*n!),n个元素的数组进行全排列的递归,每次递归都要遍历数组
  • 空间复杂度:O(n)O(n),递归栈的最大深度为数组长度n,res属于返回必要空间