31. 下一个排列

题目描述:

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

如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。

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

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

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

代码:

class Solution(object):
    def nextPermutation(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        is_possible = False
        
        # step 1: find the first nums[i] < nums[ii] from the tail
        for i in range(len(nums)-1, 0, -1):
            if nums[i-1] < nums[i]:
                i, ii = i-1, i
                is_possible = True
                break
        
        # step 2: find the first nums[j] > nums[i] from the tail and swap nums[i], nums[j]
        if is_possible:
            for j in range(len(nums)-1, -1, -1):
                if nums[j] > nums[i]:
                    nums[i], nums[j] = nums[j], nums[i]
                    # step 3: reverse all the elements starting from ii
                    nums[ii:] = nums[ii:][::-1]
                    break
        else:
            nums[:] = nums[::-1]

什么是下一个排列以及本题的思路分析:

核心思想是:

  • step 1:首先从最尾端开始往前寻找两个相邻元素,令第一元素为 i ,第二个元素为 ii ,且满足 nums[i] < nums[ii]
  • step 2:找到这样一组相邻元素后,再从最尾端开始往前检验,找出第一个大于 i 的元素,令为 j ,将 nums[i]nums[j] 对调;
  • step 3:再将 nums[ii:] 反转(reverse)。

曾出错过的地方:

  • 题目中明确说明不能返回任何东西,所以最开始写代码时,发现在VScode中能够通过,但是在LeetCode中总是报错,而且在LeetCode中的输出和在VScode中的输出不一致。后来发现是自己在代码中写了 return nums 语句,从而导致出错;
  • range左闭右开 的,在上述代码中用到了两处 range ,第一处是 range(len(nums)-1, 0, -1) ,这里右边只能取到 1,是无法取到 0 的(第二处的 range(len(nums)-1, -1, -1) 只能取到 0 而无法取到 -1),这一点在写代码时必须非常明确,否则非常容易出错。
  • 最后的 else 语句不能写成 nums = nums[::-1]
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 题目描述 实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。 如果不存在下一个更...
    youzhihua阅读 217评论 0 1
  • 题目链接难度:中等 类型: 数组 实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典...
    wzNote阅读 1,062评论 0 5
  • 更多精彩内容,请关注【力扣中等题】。 题目 难度:★★★☆☆类型:数组 实现获取下一个排列的函数,算法需要将给定数...
    玖月晴阅读 1,703评论 0 0
  • 来源:力扣(LeetCode)链接:https://leetcode-cn.com/problems/next-p...
    entre_los_dos阅读 272评论 0 1
  • 1 夜来了 我急忙找出纸和笔 要赶在 天完全黑下来之前 写下这句:我爱你 2 爱你时一切美好 怨你时... 还是爱...
    王泊一阅读 266评论 1 2

友情链接更多精彩内容