LeetCode No.31

LeetCode第三十一题

¶题目描述

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

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

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

示例 1：

输入：nums = [1,2,3]
输出：[1,3,2]
示例 2：

输入：nums = [3,2,1]
输出：[1,2,3]
示例 3：

输入：nums = [1,1,5]
输出：[1,5,1]
示例 4：

输入：nums = [1]
输出：[1]
 

提示：

1 <= nums.length <= 100
0 <= nums[i] <= 100

¶代码

class Solution(object):
    def nextPermutation(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        length = len(nums) # 总长度
        index_a = length - 1
        # 从后往前找
        while index_a >= 0:
            index_b = length - 1
            while index_b > index_a :
                # 如果大于则交换位置并对前面查找的内容进行排序
                if nums[index_b] > nums[index_a]:
                    temp = nums[index_a]
                    nums[index_a] = nums[index_b]
                    nums[index_b] = temp
                    tt = nums[index_a + 1:len(nums)]
                    tt.sort()
                    nums[index_a+1:len(nums)] = tt
                    return nums
                index_b -= 1
            index_a -= 1
        nums.reverse()
        return nums

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