Leetcode篇:下一个排列


@author: ZZQ
@software: PyCharm
@file: nextPermutation.py
@time: 2018/11/12 15:32
要求:
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
思路: 从数组尾部向前搜索,找到第一次nums[i] 比nums[i+1]小的位置,
将该位置i记录下来,然后将该位置之后的所有元素进行排列,找到比nums[i]大的第一个元素,将两数交换即可。
--------------------------------------------------------------------------------------

class Solution():
    def __init__(self):
        pass

    def quickSort(self, list, left, right):

        if left > right:
            return
        middle_value = list[left]
        i = left
        j = right

        while i < j:
            while i < j and list[j] >= middle_value:
                j -= 1
            while i < j and list[i] <= middle_value:
                i += 1

            if i < j:
                t = list[i]
                list[i] = list[j]
                list[j] = t

        list[left] = list[i]
        list[i] = middle_value
        self.quickSort(list, left, i-1)
        self.quickSort(list, i+1, right)

    def nextPermutation(self, nums):
        """
        :type nums: List[int]
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        nums_len = len(nums)
        index = 0
        small_value = -1
        if nums_len == 0:
            return nums
        for i in range(nums_len-1):
            if nums[nums_len-1-i-1] < nums[nums_len-1-i]:
                index = nums_len-1-i
                small_value = nums[nums_len-1-i-1]
                break
        if small_value == -1:
            nums.sort(reverse=False)
            return nums
        self.quickSort(nums, index, nums_len-1)
        for j in range(index, nums_len):
            if nums[j] > small_value:
                t = nums[j]
                nums[j] = nums[index-1]
                nums[index-1] = t
                break
        return nums

猜你喜欢

转载自www.cnblogs.com/zzq-123456/p/9965407.html