LeetCode 31. 下一个排列 | Python

31. 下一个排列


题目


实现获取下一个排列的函数,算法须要将给定数字序列从新排列成字典序中下一个更大的排列。python

若是不存在下一个更大的排列,则将数字从新排列成最小的排列(即升序排列)。算法

必须原地修改,只容许使用额外常数空间。数组

如下是一些例子,输入位于左侧列,其相应输出位于右侧列。bash

1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

解题思路


思路:迭代微信

首先先理解题意,题目中要求【将给定数字序列从新排列成字典序中下一个更大的排列。若是不存在,则将数字从新排列称最小的排列(即升序排列)】函数

在这里,可能直接从文字上面来看,不太不可以理解是什么意思,那么结合例子来看,先看spa

1,2,3 → 1,3,2
1,1,5 → 1,5,1

在这里,你能够理解为,要将数字 123 变为下一个更大的数字,132。115 也同理。code

而下面这个例子就是表示不存在更大的排列:blog

3,2,1 → 1,2,3

321 已是最大的了,那么就将其排列为最小的排列(升序排列),获得结果 123。索引

其实从上面的例子中,多多少少也可以看出来,在这里实际上是从后面开始找,当找到相邻升序的两个数字,在这里将它们进行交换,这样就可以获得更大的排列。

其实这里还有一部分的内容,在题目中是比较难看出来的,题目中【下一个】这个概念,其实要找到的是变化先后的排列,增长的幅度尽量小。好比,下面的例子:

1,2,3,4,5 → 1,2,3,5,4
1,2,3,5,4 → 1,2,4,3,5

第一个示例,根据上面观察所得,便是将 4 和 5 进行替换,获得更大的排列,12354。

后面的示例中 12354,获得排列的结果 12435。在这里,交换的是 3 和 4,这里其实交换的数字是尽量小的大数和前面的小数,因此并非 3 和 5 进行交换,而交换后的全部数还须要重置升序。所得出的结果是 12435,而不是 12453。

这就是关于题意的简单分析,下面看如何实现算法:

  • 首先须要明确的是从后面往前查找第一个相邻升序的两个元素所在的位置(i,j),知足 A[i] < A[j]。并且,从位置 j 日后的元素是降序的。
  • 在 j 日后的元素中,一样是从后往前找,找到第一个知足 A[i] < A[k] 的元素,将其二者进行交换。
  • 前面说了, 从 j 日后必定是降序的,那么交换之后确定也是降序的(由于找到 A[k] 是第一个比 A[i] 大的数字,因为是从后往前找,k所在位置左边的数字势必比当前 i 所在位置的数字大,而右边的数字也就比其小),这个时候,要将这部分降序,逆转为升序,这样才能保证排列先后尽量小的增幅。
  • 考虑特殊的状况,也就是整个排列是降序,便是最大的的排列时,执行上面所说逆转为升序便可。

具体的代码实现以下。

代码实现


class Solution:
    def nextPermutation(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        if len(nums) < 2:
            return

        n = len(nums)

        # 从数组右往前进行遍历,查找相邻升序元素
        i = n - 2
        j = n - 1
        while i > 0 and nums[i] >= nums[j]:
            i -= 1
            j -= 1
        # 这里有一种状况,就是循环结束后,i 为 0 且索引 0 位置的数是最大的状况
        # 那这里就表示排列就是最大的排列,将其逆转升序
        if i == 0 and nums[i]==max(nums):
            nums.reverse()
        else:
            # 当找到相邻的升序元素时
            # 再次从后往前找到一个比 nums[i] 大但相比其余元素尽量小的数
            k = n - 1
            while nums[i] >= nums[k]:
                k -= 1
            # 交换两个元素
            nums[i], nums[k] = nums[k], nums[i]

            # 如今 j 到后面的元素是降序的,这里要将其升序
            length = n - j + 1
            for x in range(length // 2):
                nums[j+x], nums[n-1-x] = nums[n-1-x], nums[j+x]

实现结果


实现结果


以上就是关于《31. 下一个排列》问题的分析及具体实现算法的主要内容。

欢迎关注微信公众号《书所集录》
相关文章
相关标签/搜索