60_第k个排列

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

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

"123"
"132"
"213"
"231"
"312"
"321"
给定 n 和 k，返回第 k 个排列。
说明：
给定 n 的范围是 [1, 9]。
给定 k 的范围是[1,  n!]。

示例 1:
输入: n = 3, k = 3
输出: "213"

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


def getPermutation(n, k):
    if n == 0:
        return []
    nums = [i+1 for i in range(n)]
    used = [False for _ in range(n)]
    return _dfs(nums, used, n, k, 0, [])

def _dfs(nums, used, n, k, depth, pre):
    if depth == n:
        return ''.join(pre)

    ps = _factorial(n-1-depth)
    for i in range(n):
        if used[i]:
            continue
        if ps<k:
            k -= ps
            continue
        pre.append(str(nums[i]))
        used[i] = True
        return _dfs(nums, used, n, k, depth+1, pre)

def _factorial(n):
    res = 1
    while n:
        res *= n
        n -= 1
    return res