LeetCode-31. 下一个排列

题目描述

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

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

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

思路

1. 从后向前遍历
2. 找到第一个正序对(nums[i-1]<nums[i])，记录i-1的位置
3. 在i到末尾这段区间中，从后往前找到第一个大于num[i-1]的值(从后往前找，可以避免从前往后找全部都是大于nums[i-1]，导致互换位置的索引值不统一)
4. 找到后，与nums[i-1]互换位置
5. 交换完位置后，从i-1之后的位置到结尾全部逆序

class Solution(object):
    def nextPermutation(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        if len(nums)<2:
            return nums
        i=len(nums)-1
        while i>0 and nums[i-1]>=nums[i]:  # 记得加等于号
            i-=1
        if i>0:
            j=len(nums)-1
            while i<=j and nums[i-1]>=nums[j]:  # 记得加等于号
                j-=1
            nums[i-1],nums[j]=nums[j],nums[i-1]
        return self.reverse(nums,i,len(nums)-1)
    def reverse(self,nums,left,right):
        while left<right:
            nums[left],nums[right]=nums[right],nums[left]
            left+=1
            right-=1
        return nums