第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”

思路:
1.先来看回溯:
依次递归全排列出,知道得到第k个排列。
比如:
n=3,k=3在这里插入图片描述

回溯得到:123 132 213停

代码:

class Solution:
    def getPermutation(self, n: int, k: int) -> str:
        
        self.m=[]
        self.j=0
        def dg(path,nn):
            if not nn:
                self.j+=1
                self.m=path[:]
                return
            for i in range(1,n+1):
                if str(i) not in path:
                    path.append(str(i))
                    dg(path,nn-1)
                    path.pop()
                    if self.j==k:#当找到第k个时,直接退出
                        return
        dg([],n)
        return "".join(self.m)

但是当n=9,就可以有9!,会有要112752全排列,这就会超时

2.再来探索全排列规律
以n=3,k=3为例

【1,2,3】均包含2!个数,3/2!=1…1
余数非零,说明这个数的位置在第二个数的全排列中,所以第一个数选nums[1],2
我们跳过了1*2!个排列,k=3-2=1,

可选【1,3】,每位数的全排列有1!个,1/1!=1…0
余数为零,则次数在第1个数的全排列里,所以第二个数选nums[1-1],1
我们跳过了(1-1)*1!=0个排列,k=1-0=1

可选【3】,每位数的全排列有0!=1个,1/0!=1…0
余数为零,则次数在第1个数的全排列里,所以第二个数选nums[1-1],3
到此3个数以排列完全,结束213

代码:

class Solution:
    def getPermutation(self, n: int, k: int) -> str:
        nums=[str(i) for i in range(1,n+1)]
        ss=''
        m=1
        for i in range(2,n):
            m*=i
        n-=1
        y,s=1,1
        while n:
            s,y=divmod(k,m)
            s=int(s)
            if s>0 and y==0:
                ss+=nums[s-1]
                nums.pop(s-1)
                k=k-(s-1)*m #当余数为零时,说明刚好在第s个数的里面,减去前s-1个全排列,注意ss+=nums[s-1]中s-1为索引,进去下一层。
            else:
                ss+=nums[s]
                nums.pop(s)
                k=y#当余数不为零时,说明在第s+1个数里边,减去前s个全排列刚好k=y,注意ss+=nums[s]中s为索引,进去下一层
            m=m/n
            n-=1
            
            
        return ss+nums[0]

而此时我们只循环了n次
所以完全不会超时

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值