leetcode60:排列序列

写在开头 转自leetcode某大佬：鲁迅曾经说过，暴力回溯超时是因为剪枝技术不好！

60. 排列序列

难度困难766收藏分享切换为英文接收动态反馈

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

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

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

给定 n 和 k，返回第 k 个排列。

示例 1：

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

示例 2：

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

示例 3：

输入：n = 3, k = 1
输出："123"

解题思路

这道题一看以为是回溯，结果简单一写超时了。
说明这样的解法是对的，只不过太慢了。
鲁迅曾经说过，暴力回溯超时是因为剪枝技术不好！
剪枝1: 搜索到第k个数的时候就直接返回答案，还是太慢了
怎么可以更快呢？
我们发现数字的排列就是以每个数字为起点,后面都是一样的！
那么我们就可以根据k和n算出来答案是以数字几为开头的，然后把这个数字作为字符串的第一个数字传入算法
之后取模就可以知道需要搜索多少次。这样可以大大减少搜索的次数！直接把复杂度降低一个数量级
最后要注意余数为0的情况，此时事实上是以上一个数开头的最后一个数

class Solution:
    def getPermutation(self, n: int, k: int) -> str:
        com = 1
        for i in range(1,n+1):
            com *= i #计算总共排列个数
        layer = com//n #计算以每个数开头的数有几个
        layers,k = (k//layer)+1,k%layer#计算第k个数在第几组的第几个
        if k == 0:#如果余数是0说明是上一层的最后一个数
            layers,k = layers-1,layer
        nums = [i for i in range(1,n+1)]
        idx = 0
        def backtrack(path,used):
            if len(path)==n:
                nonlocal ans,idx
                if idx == k: return
                ans = path
                idx += 1
                return
            for i in range(1,n+1):
                if idx == k:return
                if i in used:continue
                path += str(i)
                used.add(i)
                backtrack(path,used)
                path = path[:-1] #回溯
                used.remove(i)
        ans = '' 
        backtrack(str(layers),set([layers]))#把开头数字填答案里
        return ans

作者：jayus-x
链接：https://leetcode.cn/problems/permutation-sequence/solution/shui-shuo-bao-sou-guo-bu-liao-hui-su-jia-h11b/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。