Leetcode(1)-num31_next_permutation

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

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

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

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

"""


def nextPermutation(nums):
    """
    Do not return anything, modify nums in-place instead.
    """

    if not nums or len(nums) == 1:
        return

    i = len(nums) - 2
    while i >= 0 and nums[i] >= nums[i + 1]:
        i -= 1

    if i != -1:
        j = i + 1
        while j < len(nums) and nums[j]>nums[i]:
            j+=1
        j-=1
        nums[i],nums[j]= nums[j],nums[i]

    left = i +1
    right = len(nums)-1
    while left < right:
        nums[left],nums[right] = nums[right],nums[left]
        left +=1
        right -=1

    return nums


def main():
    print(nextPermutation([1,3,7,8,5]))



if __name__ == '__main__':
    main()

思路分析

  1. 从数组最后一个数字开始遍历,找到第一个变小的数字nums[i]
  2. 找到比num[i] 大的最小的数nums[j]
  3. 交换nums[i],nums[j]
  4. 交换从i+1开始的子数组的数字

你可能感兴趣的:(算法)