【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