(python)小菜狗算法日记（字符串系列）_leetcode 31. 下一个排列

最新推荐文章于 2022-02-27 14:14:53 发布

一小坨北北

最新推荐文章于 2022-02-27 14:14:53 发布

阅读量126

点赞数

本文链接：https://blog.csdn.net/weixin_39331401/article/details/107170684

版权

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

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

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

以下是一些例子，输入位于左侧列，其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/next-permutation

class Solution:
    def nextPermutation(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        if not nums:
            return
        lens = len(nums)
        if lens<2:
            return 
        i = lens-1
        t = lens-1
        while i>=0 and nums[i-1]>=nums[i]:
            i-=1
        while t>=0 and nums[t]<=nums[i-1]:
            t-=1
        if i==0:
            return nums.reverse()
        elif 0<i<lens-1:
            i-=1
            nums[i],nums[t]=nums[t],nums[i]
            nums[i+1:] = sorted(nums[i+1:])
            return nums[:i+1]+nums[i+1:] 
        elif i==lens-1:
            nums[lens-2],nums[lens-1]=nums[lens-1],nums[lens-2]
            return nums