回溯——46. 全排列

简介: 本专栏按照数组—链表—哈希—字符串—栈与队列—二叉树—回溯—贪心—动态规划—单调栈的顺序刷题,采用代码随想录所给的刷题顺序,一个正确的刷题顺序对算法学习是非常重要的,希望对大家有帮助

1 题目描述

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

2 题目示例

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

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/permutations

3 题目提示

1 <= nums.length <= 6
-10 <= nums[i] <= 10
nums 中的所有整数 互不相同

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

4 思路

这个问题可以看作有n个排列成一行的空格,我们需要从左往右依此填入题目给定的n个数,每个数只能使用一次。那么很直接的可以想到—种穷举的算法,即从左往右每一个位置都依此尝试填入一个数,看能不能填完这n个空格,在程序中我们可以用「回溯法」来模拟这个过程。
我们定义递归函数backtrack(first, output)表示从左往右填到第first个位置,当前排列为output。那么整个递归函数分为两个情况:

  • 如果first = n,说明我们已经填完了n个位置(注意下标从О开始),找到了一个可行的解,我们将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]。
当然善于思考的读者肯定已经发现这样生成的全排列并不是按字典序存储在答案数组中的,如果题目要求按字典序输出,那么请还是用标记数组或者其他方法。
image.png

5 我的答案

class Solution {
    public List<List<Integer>> permute(int[] nums) {
        List<List<Integer>> res = new ArrayList<List<Integer>>();

        List<Integer> output = new ArrayList<Integer>();
        for (int num : nums) {
            output.add(num);
        }

        int n = nums.length;
        backtrack(n, output, res, 0);
        return res;
    }

    public void backtrack(int n, List<Integer> output, List<List<Integer>> res, int first) {
        // 所有数都填完了
        if (first == n) {
            res.add(new ArrayList<Integer>(output));
        }
        for (int i = first; i < n; i++) {
            // 动态维护数组
            Collections.swap(output, first, i);
            // 继续递归填下一个数
            backtrack(n, output, res, first + 1);
            // 撤销操作
            Collections.swap(output, first, i);
        }
    }
}
相关文章
|
4月前
|
Kubernetes 算法 测试技术
【贪心】【回溯】【字符串】2014. 重复 K 次的最长子序列
【贪心】【回溯】【字符串】2014. 重复 K 次的最长子序列
|
3月前
|
存储 算法 数据可视化
广度优先搜索(BFS)+回溯解决LeetCode 126题:单词接龙 II
广度优先搜索(BFS)+回溯解决LeetCode 126题:单词接龙 II
|
4月前
|
算法
递归回溯解决迷宫问题
递归回溯解决迷宫问题
32 4
|
4月前
|
算法
回溯-求出数组的所有子序列【学习算法】
回溯-求出数组的所有子序列【学习算法】
44 0
|
算法 C语言
对于KMP的next数组的新发现,好像我们并不用回溯
对于KMP的next数组的新发现,好像我们并不用回溯
39 0
[leetcode] 522. 最长特殊序列 II 暴力 + 双指针
[leetcode] 522. 最长特殊序列 II 暴力 + 双指针
87 0
|
JavaScript
代码随线录刷题|LeetCode 392.判断子序列 115.不同的子序列
代码随线录刷题|LeetCode 392.判断子序列 115.不同的子序列
代码随线录刷题|LeetCode 392.判断子序列 115.不同的子序列
|
机器学习/深度学习 算法
回溯——491. 递增子序列
本专栏按照数组—链表—哈希—字符串—栈与队列—二叉树—回溯—贪心—动态规划—单调栈的顺序刷题,采用代码随想录所给的刷题顺序,一个正确的刷题顺序对算法学习是非常重要的,希望对大家有帮助
|
算法
递归+回溯求解数独问题
导读:回溯是常用的算法理论之一,很多规模较大、直接分析较为复杂的问题都可以考虑用回溯求解,例如N皇后问题、骑士周游和走迷宫问题等。本质上,回溯问题是一种优化后的暴力求解,通过及时的剪枝和启发式的寻找最优路径,可以有效加速求解过程。回溯还常常与递归搭配使用
147 0
递归+回溯求解数独问题