【刷题】LC-60：排列序列

题目描述：

难度：hard。

普通回溯+剪枝易理解版本：

 List<String> res = new ArrayList<>();
    boolean [] used;
    int resCount = 0;
    public String getPermutation(int n, int k) {
        used = new boolean[n + 1];
        getPermutation(n,k,"");
        return res.get(k - 1);
    }
    private void getPermutation(int n,int k,String aRes){
        if(aRes.length() == n){
            res.add(aRes);
            resCount++;
            return;
        }
        for(int i = 1;i <= n;i++){
            if(resCount == k)
                return;
            if(used[i])
                continue;
            used[i] = true;
            getPermutation(n,k, aRes + String.valueOf(i));
            used[i] = false;
            
        }
    }

数学法剪枝：

import java.util.Arrays;

public class Solution {

    /**
     * 记录数字是否使用过
     */
    private boolean[] used;

    /**
     * 阶乘数组
     */
    private int[] factorial;

    private int n;
    private int k;

    public String getPermutation(int n, int k) {
        this.n = n;
        this.k = k;
        calculateFactorial(n);

        // 查找全排列需要的布尔数组
        used = new boolean[n + 1];
        Arrays.fill(used, false);

        StringBuilder path = new StringBuilder();
        dfs(0, path);
        return path.toString();
    }


    /**
     * @param index 在这一步之前已经选择了几个数字，其值恰好等于这一步需要确定的下标位置
     * @param path
     */
    private void dfs(int index, StringBuilder path) {
        if (index == n) {
            return;
        }

        // 计算还未确定的数字的全排列的个数，第 1 次进入的时候是 n - 1
        int cnt = factorial[n - 1 - index];
        for (int i = 1; i <= n; i++) {
            if (used[i]) {
                continue;
            }
            if (cnt < k) {
                k -= cnt;
                continue;
            }
            path.append(i);
            used[i] = true;
            dfs(index + 1, path);
            // 注意 1：不可以回溯（重置变量），算法设计是「一下子来到叶子结点」，没有回头的过程
            // 注意 2：这里要加 return，后面的数没有必要遍历去尝试了
            return;
        }
    }

    /**
     * 计算阶乘数组
     *
     * @param n
     */
    private void calculateFactorial(int n) {
        factorial = new int[n + 1];
        factorial[0] = 1;
        for (int i = 1; i <= n; i++) {
            factorial[i] = factorial[i - 1] * i;
        }
    }
}