【字节专题】leetcode 60. 第k个排列

60. 第k个排列

给出集合 [1,2,3,…,n]，其所有元素共有 n! 种排列。

按大小顺序列出所有排列情况，并一一标记，当 n = 3 时, 所有排列如下：

"123"
"132"
"213"
"231"
"312"
"321"
给定 n 和 k，返回第 k 个排列。

说明：
给定 n 的范围是 [1, 9]。
给定 k 的范围是[1,  n!]。

示例 1:
输入: n = 3, k = 3
输出: "213"

示例 2:
输入: n = 4, k = 9
输出: "2314"

第一想法是修改一下46. 全排列代码，加个计数器数到第k个就返回，虽然能过，但是时间效率很差。
那么这里其实可以加剪枝，并不需要都遍历到叶子节点，很多都是可以跳过的，以n = 4, k = 9为例，第一层，以1开头，后面有3个数“2、3、4”可选，以1开头的全排列一共有3!=6个，并且6<k=9，因此，所求的全排列一定不以1开头，可以在这里“剪枝”。后面以此类推。

代码写得有点丑不管了。。

int k;
String ans;

public static void main(String[] args) {
    String ret = new 第k个排列().getPermutation(4,9);
    System.out.println(ret);
}
public String getPermutation(int n, int k) {
    this.k = k;
    permutation(n, "", new boolean[n+1]);
    return ans;
}

private void permutation(int n, String str, boolean[] visited){
    if (str.length()==n){
        k--;
        if (k==0){
            ans = str;
        }
        return;
    }
    int ps = factorial(n-1-str.length());
    for (int i = 1;i<=n;i++){
        if (visited[i]) continue;
        if (ps<k){
            k-=ps;
            continue;
        }
        visited[i] = true;
        permutation(n, str+String.valueOf(i), visited);
        visited[i] = false;
        if (ans!=null)    return;
    }
}

private int factorial(int n){
    int res = 1;
    while (n>0){
        res *= n;
        n--;
    }
    return res;
}