[LeetCode]60 第k个排列

题目描述

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

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

  1. "123"
  2. "132"
  3. "213"
  4. "231"
  5. "312"
  6. "321"

给定 nk,返回第 k 个排列。

说明:

  • 给定 n 的范围是 [1, 9]。
  • 给定 k 的范围是[1, n!]。

示例1:

输入: n = 3, k = 3
输出: "213"

示例2:

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

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
class Solution:
def getPermutation(self, n, k):
"""
:type n: int
:type k: int
:rtype: str
"""
rt = []
k = k - 1
def pushNum(nums,k):
if nums:
flag = int(k/math.factorial(len(nums)-1))
rt.append(nums[flag])
pushNum(nums[:flag] + nums[flag+1:], k - flag * math.factorial(len(nums)-1))

pushNum([str(i) for i in range(1, n+1)], k)
return ''.join(rt)