LeetCode Python - 60. 排列序列

---

题目描述

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

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

1. “123”
2. “132”
3. “213”
4. “231”
5. “312”
6. “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!

---

解法

枚举

我们知道，集合 [1,2,…n] 一共有 n! 种排列，如果我们确定首位，那剩余位能组成的排列数量为 (n−1)!。

因此，我们枚举每一位 i，如果此时 k 大于当前位置确定后的排列数量，那么我们可以直接减去这个数量；否则，说明我们找到了当前位置的数。

对于每一位 i，其中 0≤i<n，剩余位能组成的排列数量为 (n−i−1)!，我们记为 fact。过程中已使用的数记录在 vis 中。

时间复杂度 O(n ² )，空间复杂度 O(n)。

class Solution(object):
    def getPermutation(self, n, k):
        """
        :type n: int
        :type k: int
        :rtype: str
        """
        ans = []
        vis = [False] * (n + 1)
        for i in range(n):
            fact = 1
            for j in range(1, n - i):
                fact *= j
            for j in range(1, n + 1):
                if not vis[j]:
                    if k > fact:
                        k -= fact
                    else:
                        ans.append(str(j))
                        vis[j] = True
                        break
        return ''.join(ans)

---

运行结果