60-下一个排列-LeetCode31（python）

• 题目描述

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

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

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

• 示例

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

• 思路

参照官方题解的思路。

（1）首先从后往前遍历，找到第一个升序对nums[i-1]和nums[i]，以索引 i-1对应的元素为分界线，它与它右边的元素均需要进行处理。

（2）此时nums[i-1]后边的元素必然是降序排列的，那么就再一次从后往前进行遍历，找到第一个大于nums[i-1]的元素nums[j]，就是要和nums[i-1]交换的元素了。将nums[i-1]和nums[j]交换之后，nums[i-1]右侧的序列其实仍不满足我们的要求，因为这个子序列还是降序排列的，不是其构成元素所能组成的最小字典序列，对该子序列进行升序排列，即可满足要求。

（3）需要注意的是，有可能给定的nums本身就是个降序排列的数组，那么在首次遍历时就找不到满足条件的nums[i-1]和nums[i]，也就是说上述这些操作都不会进行，对这样的nums输入而言，题目要求的输出是升序排列的nums，所以要对这种情况进行特殊处理。

• 代码

class Solution:
    def nextPermutation(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        #从后往前遍历，寻找第一个num[i-1]<nums[i]的元素对
        for i in range(n-1,0,-1):
            #存在比当前序列大的字典序
            #找到了nums[i-1]和nums[i]
            if nums[i-1] < nums[i]:
                #再次从后往前遍历，寻找刚刚比nums[i-1]大的数
                #因此此时num[i-1]右侧的序列为降序序列，那么从后往前遍历到的第一个大于nums[i-1]的元素就是要找的元素nums[j]
                for j in range(n-1,i-1,-1):
                    #找到了nums[j]
                    if nums[j] > nums[i-1]:
                        #交换nums[j]和nums[i-1]
                        nums[j],nums[i-1] = nums[i-1],nums[j]
                        
                        #对nums[i-1]右侧的子序列重新排序为升序
                        left,right = i, n-1
                        while left < right:
                            nums[left],nums[right] = nums[right],nums[left]
                            left += 1
                            right -= 1
                        return
        #输入的nums为降序序列
        #即不存在比当前序列大的字典序，那么返回的是整个nums的升序排列
        nums.sort()