分类:array
难度:medium
- 下一个排列
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
题解
考虑字典序的排列,首先我们从右往左遍历,如何a[i]>a[i-1],那么首先a[i]是需要被替代的,被谁替代呢?
此时a[i:]是一个单调递减(准确来说非递增)的序列,找出a[i:]中比a[i-1]大的最小数,然后与a[i-1]交换,此时a[i:]仍是一个递减序列,逆转之后,变成了进位之后最小的排列
原题解在这
https://leetcode-cn.com/problems/next-permutation/solution/xia-yi-ge-pai-lie-by-leetcode/
代码
class Solution(object):
def nextPermutation(self, nums):
"""
:type nums: List[int]
:rtype: None Do not return anything, modify nums in-place instead.
"""
for i in range(len(nums)-1,0,-1):
if nums[i]>nums[i-1]:
tag = nums[i-1]
j = i
while j<len(nums):
if nums[j]>tag:
j+=1
else:
break
# j-1 是大于tag的最小的数
nums[i-1] = nums[j-1]
nums[j-1] = tag
left,right = i,len(nums)-1
# print(nums)
while left<right:
tmp = nums[left]
nums[left]=nums[right]
nums[right] = tmp
left+=1
right-=1
return nums
left,right = 0,len(nums)-1
while left<right:
tmp = nums[left]
nums[left]=nums[right]
nums[right] = tmp
left+=1
right-=1
return nums
这个代码写的有点丑了