31. 下一个排列
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。1,2,3
→ 1,3,2
3,2,1
→ 1,2,3
1,1,5
→ 1,5,1
解题思路:
第一步:从前向后找到最后的升序索引index1
第二步:从index1+1向后找到最后的较大值索引index2
第三步:交换index1与index2的元素
第四步:对nums[index1+1:]进行翻转
Python3代码如下:
class Solution(object):
def nextPermutation(self, nums):
"""
:type nums: List[int]
:rtype: None Do not return anything, modify nums in-place instead.
"""
index1,index2 = -1,0
for i in range(len(nums)-1):
if nums[i] < nums[i+1]:
index1 = i
if index1 == -1:
nums.reverse()
return
for i in range(index1+1,len(nums)):
if nums[i] > nums[index1]:
index2 = i
nums[index1],nums[index2] = nums[index2],nums[index1]
nums[index1+1:] = nums[:index1:-1]