LeetCode31. 下一个排列
Difficulty: 中等
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
示例 1:
输入:nums = [1,2,3]
输出:[1,3,2]
示例 2:
输入:nums = [3,2,1]
输出:[1,2,3]
示例 3:
输入:nums = [1,1,5]
输出:[1,5,1]
示例 4:
输入:nums = [1]
输出:[1]
Solution
分三个步骤
- 从后往前找到第一个不是降序的下标
- 倒着找到第一个比它大的,交换
- 反转后边列表(因为后边就是降序,反过来就完了。如果遇到整个列表降序的情况,全部倒过来就完了)
class Solution(object):
def nextPermutation(self, nums):
"""
:type nums: List[int]
:rtype: None Do not return anything, modify nums in-place instead.
"""
p=-1
q=-1
#步骤1:从后往前找到第一个不是降序的下标
for i in range(len(nums)-2,-1,-1):
if nums[i]<nums[i+1]:
p=i
break
#步骤2:倒着找到第一个比它大的,交换
if p!=-1:
for i in range(p,len(nums)):
if nums[p]<nums[i]:
q=i
nums[p],nums[q]=nums[q],nums[p]
#原来使用 nums=nums[:p+1]+sorted(nums[p+1:]) 错了,原因:必须原地修改!
l=p+1
r=len(nums)-1
while l<r:#步骤三:反转后边列表(因为后边就是降序,反过来就完了
nums[l],nums[r]=nums[r],nums[l]
l+=1
r-=1
else:#步骤三(整个列表降序的情况)全部倒过来就完了
l=0
r=len(nums)-1
while l<r:
nums[l],nums[r]=nums[r],nums[l]
l+=1
r-=1