leetcode_31_下一个排列

下一个排列

描述

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

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

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

示例

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

解题

题目看的有懵，这是干啥，看了评论才发现
让我们找出这几个数中排序出的所有的数中，刚好比当前数大的数
如果已经是最大的了，那就返回最小的数

如果一个数是最小的数，那么组成这个数的数字肯定是依次增大的，比如123，234
如果一个数是最大的数，那么组成这个数的数字肯定是依次减小的，比如321，432，变成逆序就变成了下一个数，也就是最小的数

如果既不是最大的数，也不是最小的数，如14532
从后往前找第一次正序变大的两个数，如1和4，将后面的数反转即为依次增大的
再将4和后面大于4且最小的数5交换得15234

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

        # 数字全为逆序，最大值，反转后为下一个值，最小值
        if k == -1:
            nums.reverse()
            return 
        nums[k+1:] = nums[:k:-1]

        for i in range(k+1, len(nums)):
            if nums[k] < nums[i]:
                nums[k], nums[i] = nums[i], nums[k]
                break 


if __name__ == '__main__':
    nums = [1,2,3]
    Solution().nextPermutation(nums)
    print(nums)

    Solution().nextPermutation(nums)
    print(nums)

    Solution().nextPermutation(nums)
    print(nums)

    Solution().nextPermutation(nums)
    print(nums)

    Solution().nextPermutation(nums)
    print(nums)