leetcode 031. 下一个排列

2019-01-01  本文已影响11人  阳光树林

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

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

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

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

思路:
124653 --> 125346
125430 --> 130245
从右向左找,只有找到前一位nums[i-1]小于当前位nums[i]的,排列后才可能比当前数字大;将nums[i-1]和右边找到的刚好大于nums[i-1]的数字作交换,最后把交换后的nums[i:]从小到大排列,保证结果最小,即nums[i:]倒序,因为交换后nums[i:]仍是从大到小。

class Solution:
    def nextPermutation(self, nums):
        n = len(nums)
        for i in range(n - 1, 0, -1):
            if nums[i-1] < nums[i]:
                for j in range(n-1, i-1, -1):
                    if nums[j] > nums[i-1]:
                        nums[j], nums[i-1] = nums[i-1], nums[j]
                        nums[i:] = nums[i:][::-1]
                        return
        else:
            nums.reverse()

su = Solution()
nums = [3,2,1]
su.nextPermutation(nums)
print(nums)
class Solution(object):
    def nextPermutation(self, nums):
        """        :type nums: List[int]        :rtype: void Do not return anything, modify nums in-place instead.        """
        if len(nums) <= 1:
            return
        idx = 0
        for i in range(len(nums)-1, 0, -1):
            if nums[i] > nums[i-1]: # find first number which is smaller than it's after number
                idx = i
                break
        if idx != 0: # if the number exist,which means that the nums not like{5,4,3,2,1}
            for i in range(len(nums)-1, idx-1, -1):
                if nums[i] > nums[idx-1]:
                    nums[i], nums[idx-1] = nums[idx-1], nums[i]
                    break

        nums[idx:] = nums[idx:][::-1]
上一篇下一篇

猜你喜欢

热点阅读