[LeetCode]31、下一个排列

题目描述

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

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

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

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

思路

1、先找出最大的索引，满足nums[k] < nums[k+1],找不到，则反转数组
2、再找出另一个最大的索引，满足nums[l] > nums[k]
3、交换两者
4、翻转nums[k+1:]

class Solution:
    def nextPermutation(self, nums):
        if not nums or len(nums) == 0:
            return 0
        first_k = -1
        for i in range(len(nums) - 2, -1, -1):
            if nums[i] < nums[i + 1]:
                first_k = i
                break
        if first_k != -1:
            second_k = -1
            for i in range(len(nums) - 1, first_k, -1):
                if nums[i] > nums[first_k]:
                    second_k = i
                    break
            nums[first_k], nums[second_k] = nums[second_k], nums[first_k]
        i = first_k + 1
        j = len(nums) - 1
        while i < j:
            nums[i], nums[j] = nums[j], nums[i]
            i += 1
            j -= 1

AC31