美文网首页
leetcode 031. 下一个排列

leetcode 031. 下一个排列

作者: 阳光树林 | 来源:发表于2019-01-01 19:48 被阅读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]
    

    相关文章

      网友评论

          本文标题:leetcode 031. 下一个排列

          本文链接:https://www.haomeiwen.com/subject/utkolqtx.html