leetcode 31. 下一个排列

Posted 2021-02-05 树的种子

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

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

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

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

 

先判断是否全是逆序；是逆序只要倒序一下就行。

否则从有往左，找到第一个顺序，说明那里需要修改，然后从后面挑一个最小的大于他的数，交换位置，然后把后面排序就行。

class Solution:
    def nextPermutation(self, nums):
        """
        :type nums: List[int]
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        flag = -1
        for i in range(len(nums)-1,0,-1):
            if nums[i-1] < nums[i]:
                flag = i
                break

        if flag==-1:
            nums.sort()
        else:

            pos = flag
            minn = nums[flag]
            for i in range(flag, len(nums)):
                if minn > nums[i] and nums[i] > nums[flag-1]:
                    minn = nums[i]
                    pos = i

            nums[flag-1] , nums[pos] = nums[pos] , nums[flag-1]
            nums[flag:] = sorted(nums[flag:])