leetcode之Next Permutation(31)

题目:

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

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

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

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

Python代码:

class Solution:
    def nextPermutation(self, nums):
        partition = -1
        for i in range(len(nums) - 2, -1, -1):
            if nums[i] < nums[i+1]:
                partition = i
                break
        if partition == -1:
            nums.reverse()
        else:
            for i in range(len(nums) - 1, partition,-1):
                if nums[i] > nums[partition]:
                    nums[i], nums[partition] = nums[partition], nums[i]
                    break
            nums[partition:] = nums[partition:][::-1]

心得:原地操作,leetcode上无需返回,另外最后一步逆序操作是为了找到其下一个排列,因为partition后的几位为降序数,这几个数组成的最小的数应该为升序数。

猜你喜欢

转载自blog.csdn.net/cuicheng01/article/details/82290435
今日推荐