排列型回溯

相比于组合，排列型回溯对于元素的顺序是有要求的。
在这里插入图片描述
为了告诉回溯下面还可以选择哪些数字，可以：

记录已经被选择的数字
用一个集合存储还可以被选择的数字

对于排列型回溯，（与其他类型回溯的主要区别就在于 需要额外的记录）

例题1——46. 全排列

https://leetcode.cn/problems/permutations/
在这里插入图片描述

依次尝试各个位置上放置哪个数字，同时记录哪些数字是已经被放置过了的。

class Solution {
    List<List<Integer>> ans = new ArrayList();
    List<Integer> t = new LinkedList();

    public List<List<Integer>> permute(int[] nums) {
        int n = nums.length;
        boolean[] st = new boolean[n];  // 记录第i个数是否已经被选过
        dfs(nums, st, 0);   // 第0个位置放置哪个数字
        return ans;
    }

    public void dfs(int[] nums, boolean[] st, int k) {
        if (k == nums.length) {
            ans.add(new ArrayList(t));
            return;
        }
        for (int i = 0; i < nums.length; ++i) { // 依次尝试各个数字放在第k个位置
            if (st[i]) continue;
            st[i] = true;
            t.add(nums[i]);
            dfs(nums, st, k + 1);
            st[i] = false;	// 恢复现场
            t.remove(t.size() - 1);
        }
    }
}

除了额外声明一个 st数组来记录每个数字的状态之外，还可以通过给 nums数组中已经被选择的数字加上一个特别大的数字，这样如果一个数字超过了原本数据的范围，那么就表示它已经被选择过了。

例题2——N皇后

https://leetcode.cn/problems/n-queens/

在这里插入图片描述

class Solution {
    char[][] borad;
    Set<Integer> s1 = new HashSet(), s2 = new HashSet(), s3 = new HashSet();
    List<List<String>> ans = new ArrayList();

    public List<List<String>> solveNQueens(int n) {
        borad = new char[n][n];
        for (int i = 0; i < n; ++i) Arrays.fill(borad[i], '.');
        dfs(0, n);
        return ans;
    }   

    public void dfs(int row, int n) {
        if (row == n) {
            List<String> t = new ArrayList();
            for (char[] line: borad) t.add(new String(line));
            ans.add(t);
            return;
        }
        for (int i = 0; i < n; ++i) {   // 遍历这一行的每一列
            if (!s1.contains(i) && !s2.contains(i + row) && !s3.contains(i - row)) {
                s1.add(i);
                s2.add(i + row);
                s3.add(i - row);
                borad[row][i] = 'Q';
                dfs(row + 1, n);
                borad[row][i] = '.';
                s1.remove(i);
                s2.remove(i + row);
                s3.remove(i - row);
            }
        }
    }
}