题目描述
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
思路
- 从后向前遍历
- 找到第一个正序对(nums[i-1]<nums[i]),记录i-1的位置
- 在i到末尾这段区间中,从后往前找到第一个大于num[i-1]的值(从后往前找,可以避免从前往后找全部都是大于nums[i-1],导致互换位置的索引值不统一)
- 找到后,与nums[i-1]互换位置
- 交换完位置后,从i-1之后的位置到结尾全部逆序
class Solution(object):
def nextPermutation(self, nums):
"""
:type nums: List[int]
:rtype: None Do not return anything, modify nums in-place instead.
"""
if len(nums)<2:
return nums
i=len(nums)-1
while i>0 and nums[i-1]>=nums[i]: # 记得加等于号
i-=1
if i>0:
j=len(nums)-1
while i<=j and nums[i-1]>=nums[j]: # 记得加等于号
j-=1
nums[i-1],nums[j]=nums[j],nums[i-1]
return self.reverse(nums,i,len(nums)-1)
def reverse(self,nums,left,right):
while left<right:
nums[left],nums[right]=nums[right],nums[left]
left+=1
right-=1
return nums