【Python成长之路】极简代码 -- 第K个排列

【LeetCode试题】

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

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

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

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

 

【分析】

常见的解题思路是回溯法、递归等，但是不够pythonic。可以直接用python自带的排列函数

# coding=utf-8
import itertools
def getPernumtation(n,k):
    list1 = range(1,n+1)
    templist = list(itertools.permutations(list1))
    for i,temp in enumerate(templist):
        if k == i+1:
            return ''.join([str(j) for j in temp])
print(getPernumtation(3,3))

permutations：对列表元素进行排列，如果添加上第2个参数，可以规定排列元素个数

combinations：对字符串元素进行组合，如果添加上第2个参数，可以规定组合元素个数。用法和permutations一样。

 

【参考】

https://blog.csdn.net/mishi_zcf/article/details/52455688