Leetcode60、排列序列

一、题目

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

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

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

示例 1：

输入：n = 3, k = 3
输出：“213”
示例 2：

输入：n = 4, k = 9
输出：“2314”
示例 3：

输入：n = 3, k = 1
输出：“123”

提示：

1 <= n <= 9
1 <= k <= n!
通过次数79,112提交次数151,993

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/permutation-sequence
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、代码实现

bug积累:
map表的get在没有该key值时会返回null，如果这个返回值null做算术运算，就会出现空指针异常

程序思想积累:
1、预处理数组
2、回溯中的剪枝

class Solution {
     //存储阶乘结果
       static  HashMap<Integer, Integer> map = new HashMap<>();
        static{for (int i = 0; i <= 9; i++) {
            map.put(i, Factorial(i));
        }}
     public String getPermutation(int n, int k) {
        StringBuilder ans = new StringBuilder();
        //mark数组为预处理数组，是用来是否已用，以及如果未用当前位置在所有位置的相对postion
        int[] mark = new int[n + 1];
        for (int i = 1; i < n + 1; i++) {
            mark[i] = i - 1;
        }
        resolve(ans, 1, n, k - 1, mark);
        return ans.toString();
    }


    private static Integer Factorial(int i) {
        if (i == 0) return 0;
        if (i == 1) return 1;
        return i * Factorial(i - 1);
    }

    private void resolve(StringBuilder ans, int pos, int n, int k, int[] mark) {
        if (pos == n + 1) return;
        for (int i = n; i >= 1; i--) {
            int curK = k - mark[i] * map.get(n - pos);
            if (mark[i] < 0 || curK < 0) continue;
            ans.append(i);
            //mark为负表示已经加入ans
            mark[i] = -1;
            for (int j = i + 1; j <= n; j++) {
                mark[j]--;
            }
            resolve(ans, pos + 1, n, curK, mark);
            break;
        }
    }
}

三、代码优化

class Solution {
    static int[] factorial = new int[10];
    static {
        factorial[0] = 1;
        for (int i = 1; i <= 9; i++) {
            factorial[i] = i * factorial[i - 1];
        }
    }
    public String getPermutation(int n, int k) {
        StringBuilder ans = new StringBuilder();
        List<Integer> list = new ArrayList<>();
        for (int i = 1; i <= n; i++) {
            list.add(i);
        }
        k--;
        for (int i = n - 1; i >= 0; i--) {
            int index = k / factorial[i];
            ans.append(list.remove(index));
            k %= factorial[i];
        }
        return ans.toString();
    }
}