实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
- 从后往前遍历,每一次判断nums[i]是否大于nums[i+1],如果大于就继续,不大于就暂时停止。
- 遍历一遍得到i之后,如果i==-1,说明整个数组都是递减的,此时的下一个数值为整个数组的reverse
- 如果i在中间的某一个idx,从i往后遍历,找到最小的那个大于nums[i]的数,swap这两个数
- reverse之后,将i后面的整个数组reverse
举个例子:
1 2 3 4 5 6 7
| 1 2 5 7 6 4 3 i j swap: 1 2 6 7 5 4 3 i reverse: 1 2 6 3 4 5 7
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
| class Solution(object): def nextPermutation(self, nums): """ :type nums: List[int] :rtype: None Do not return anything, modify nums in-place instead. """ def reverse(nums, left, right): while left < right: nums[left], nums[right] = nums[right], nums[left] left += 1 right -= 1 if not nums: return i = len(nums)-2 while i >= 0 and nums[i] >= nums[i+1]: i -= 1 if i == -1: nums.reverse() return j = i+1 while j <= len(nums)-1 and nums[j]>nums[i]: j += 1 j -= 1 nums[i], nums[j] = nums[j], nums[i] reverse(nums, i+1, len(nums)-1)
|