31-下一个排列

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

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

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

示例

1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

解法

核心思想：

先找出最大的索引 k 满足 nums[k] < nums[k+1]，如果不存在，就翻转整个数组；
再找出另一个最大索引 l 满足 nums[l] > nums[k]；
交换 nums[l] 和 nums[k]；
最后翻转 nums[k+1:]。

举个例子：

比如 nums = [1,2,7,4,3,1]，下一个排列是什么？

我们找到第一个最大索引是 nums[1] = 2，则firstIndex=1

再找到第二个最大索引是 nums[4] = 3，则secondIndex=4

交换nums[1]和nums[4]，nums = [1,3,7,4,2,1];

翻转[firstIndex+1:]，nums = [1,3,1,2,4,7] ，完毕!

class Solution:
    def nextPermutation(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        # 初始化两个索引值
        firstIndex, secondIndex = -1, -1
        n = len(nums)
        # 找第一个最大索引
        for i in range(n-2, -1, -1):
            if nums[i] < nums[i+1]:
                firstIndex = i
                break
        # 未找到则反转整个数组
        if firstIndex == -1:
            # nums[:]为使得nums的地址不变
            nums[:]=nums[::-1]
            return 
        # 找第二个索引
        for i in range(n-1, firstIndex, -1):
            if nums[i] > nums[firstIndex]:
                secondIndex = i
                break
        nums[firstIndex], nums[secondIndex] = nums[secondIndex], nums[firstIndex]
        nums[:]= nums[:firstIndex+1]+nums[firstIndex+1:][::-1]

来源

LeetCode中该题地址，Click here!