回溯DFS

目录

46. 全排列

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

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

解:
1.对与DFS的题 画出下面这个图
回溯DFS
2.设答案数组为ans ans中包含的一个个数组用arr标志
3.左边这个黑框第一列代表标志数组 用来记录这个数是否在arr中 第二列代表需要进行全排列的数组 本题解是[5,6,7] 本题是[1,2,3] 第三列代表当前遍历到哪一层 当层数=数组长度时 将arr加入ans中
4.由于本题数字不能重复使用 即不能出现[5,5,5] 这种数组 所以我们需要一个标志数组来记录 哪些数字加入了arr
5.DFS代码套路:
(1)先确定回溯条件:当层数=数组长度时 相当于遍历完了 完成了一次全排列 将arr加入ans中
(2)DFS代码:因为要全排列 所以第一步肯定是遍历数组 判断该数字是否已经在arr中 若不在 则加入 更改标志位 进入下一层 调用DFS
(3)执行完dfs(nums, flags, index, ans, arr);这行代码代表你已经完成一次全排列了 现在需要回溯:将刚才设置的标志位全部复原:从arr中弹出元素 复原标志位 改变层数
6.注意将arr添加到ans时 需要拷贝当前的arr添加到ans:arr.slice() 因为你后面还要对arr进行操作 比如回溯时会移除最后一个元素 那么ans中的arr也会跟着变 所以这里一定要记得拷贝!
代码如下:
JS:

var permute = function(nums) {
    let flags = [];
    for (let i = 0; i < nums.length; i++) {
        flags.push(false);
    }
    let ans = [];
    let arr = [];
    dfs(nums, flags, 0, ans, arr);
    return ans;
};
function dfs(nums, flags, index, ans, arr) {
    if (index === nums.length) {
        ans.push(arr.slice());
        return;
    }
    for (let i = 0; i < nums.length; i++) {
        if (!flags[i]) {
            arr.push(nums[i]);
            flags[i] = true;
            index++;
            dfs(nums, flags, index, ans, arr);
            arr.pop();
            flags[i] = false;
            index--;
        }
    }
}

也可以用set代替flags 代码如下:但是不建议这样做 因为在数据较少情况下 用数组比用set好

var permute = function(nums) {
        let flags = new Set();
        let ans = [];
    let arr = [];
    dfs(nums, flags, 0, ans, arr);
    return ans;
};
function dfs(nums, flags, index, ans, arr) {
    if (index === nums.length) {
        ans.push(arr.slice());
        return;
    }
    for (let i = 0; i < nums.length; i++) {
        if (!flags.has(i)) {
            arr.push(nums[i]);
            flags.add(i);
            index++;
            dfs(nums, flags, index, ans, arr);
            arr.pop();
            flags.delete(i);
            index--;
        }
    }
}

JAVA:

public List<List<Integer>> permute(int[] nums) {
        List<List<Integer>> lists = new LinkedList<>();
        List<Integer> list = new LinkedList<>();
        boolean[] used = new boolean[nums.length];
        dfs(nums,0,lists,list,used);
        return lists;
    }

    private static void dfs(int[] nums,int i,List<List<Integer>> lists,List<Integer> list,boolean[] used){
        if (i == nums.length) {
            lists.add(new LinkedList<>(list));
            return;
        }
        for (int j = 0; j < nums.length; j++) {
            int num = nums[j];
            if (!used[j]){
                used[j] = true;
                list.add(num);
                dfs(nums,i + 1,lists,list,used);
                used[j] = false;
                list.remove(list.size() - 1);
            }
        }
    }
上一篇:网络编程TCP—2


下一篇:QComboBox某项不可选中方法