LeetCode-python 60.第k个排列

题目链接
难度: 中等       类型:数学


给出集合 [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"

解题思路


对于n个数的排列,以‘1’开头的排列有(n-1)!个,以‘2’开头的排列有(n-1)!个,等等,通过这种方式可以计算出第k个排列是由哪个数字开头的

例如:
n=3, k=3
以‘1’开头的排列有2 (由(3-1)!计算可知)个,第3个,是由‘2’开头,现在数字只剩下‘1’,‘3’
问题转化为‘1’,‘3’两个数构成的排列中,第1(由3-2得)个排列的第1个数字是多少,依次类推,确定所有数字

代码实现

import math
class Solution:
    def getPermutation(self, n: int, k: int) -> str:
        kth = ''
        nums = list(map(str, range(1, n+1)))
        k -= 1
        while n>0:
            n -= 1
            f = math.factorial(n)
            index, k = divmod(k, f)
            kth += nums.pop(index)        
        return kth

你可能感兴趣的:(LeetCode-python 60.第k个排列)