LeetCode.31. 下一个排列（全排列）

1、题目描述

https://leetcode-cn.com/problems/next-permutation/

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

如果不存在下一个更大的排列，则将数字重新排列成最小的排列（即升序排列）。必须原地修改，只允许使用额外常数空间。

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

123 456
123 465
123 546
123 564
123 645
123 654
124 356
124 365
124 536
124 563
...
654 321

2、代码详解

1. 从后往前寻找第一个升序对(i,j)，即nums[i]<nums[j]，i-1位置为小数
2. 再从后往前找第一个大于nums[i]的数，即为大数，j位置
3. 交换着两个元素，即将大数换到前面
4. ​​​​​​​然后将大数后面的部分倒序

class Solution:
    def nextPermutation(self, nums):
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        if n < 2:
            return nums
        i = n - 1
        # （1）寻找第一个升序对，i-1为小数。（第一遍出错就是这儿没有等于）
        while i > 0 and nums[i-1] >= nums[i]:
            i -= 1
        # i-1为小数！

        if i == 0:  # 特殊情况：此数为最大数，未找到升序对（降序）
            # nums不存在下一个更大的排列，直接转为最小的排列（升序）
            # （之前写的 i==0 and nums[i]==max(nums)，判断冗余现删除）
            # print(nums.reverse())  # 返回 None
            nums.reverse()
            return nums
        else:
            j = n - 1

            while j > i-1 and nums[j] <= nums[i-1]:  # （2）从后往前找第一个大于nums[i-1]的数
                j -= 1
            # j为大数！
            # （3）交换着两个元素，即将大数换到前面
            nums[i-1], nums[j] = nums[j], nums[i-1]
            # （4）然后将大数后面的部分倒序
            for k in range((n-i) // 2):
                nums[i+k], nums[n-1-k] = nums[n-1-k], nums[i+k]

            return nums  # 原地修改

测试

'''
123
132
213
231
312
321
'''

nums = [1,2,4,3,6,5]
s = Solution()
print(s.nextPermutation(nums))