Leetcode 31 下一个排列 python

描述

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

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

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

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

分析

从最后一个开始找，找到第一个非升序的数，记录其位置为i
然后从i+1后面的数中，找到最后边一个比nums[i]大的数
两者交换
然后将位置i后面的数字转置一下（因为本来就是升序的，现在转置就变成降序了）

代码

class Solution:
    def swap(self,s1,s2):
        return s2,s1
    def nextPermutation(self, nums) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        llen = len(nums)

        i = llen-2
        j = llen-1
        while i>=0 and nums[i]>=nums[i+1]:
            i -= 1
        if i>=0:
            while nums[j] <= nums[i]:
                j -= 1
            nums[i],nums[j] = self.swap(nums[i],nums[j])
        nums[i+1:llen] = reversed(nums[i+1:llen])