题目描述
实现获取 下一个排列 的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。必须 原地 修改,只允许使用额外常数空间。
例子:
输入:nums = [1,2,3]
输出:[1,3,2]
题目解析
乍一看题目都不懂具体意思。一步一步来分析,首先是字典序列,就是1,3,4,5,6……9,a,b,c……z(该题目中只包含数字)。学过排列组合的同学都知道,1,2,3的全排列有:
123
132
213
231
312
321
并且在最开始用列举法来进行排列组合的时候,老师就有说“列举要从小到大来,不容易遗漏”。所以就如题目所述,123的后一个排列为132,231的后一个排列为321。
解题思路
Step 1:从后往前遍历整个数组,找到第一个不是升序排列的数字,比如46532,第一个不是升序排列的数字就为4.
Step 2:从后网前遍历到Step 1找到的“4”之前的所有数字,就是532,在这三个数中找到第一个比4大的元素“5”,交换这两个数字,现在的数组就变成了56432,但现在还不是最接近46532的数字
Step 3:再对Step 1中“4”对应的下标之后的数字进行reverse排列,所以就变成了52346。
class Solution:
def nextPermutation(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
lenth = len(nums)
j = lenth - 1
i = lenth - 1
# 从后往前找,找到第一个不是降序排列的数,nums[j-1]
while(nums[j-1] >= nums[j] and j >= 1):
j = j - 1
# 从后往前找,找到第一个大于nums[j-1]的数字
if j == 0:
# 直接reverse
l = j
r = lenth-1
while(l <= r):
nums[l],nums[r] = nums[r],nums[l]
l = l + 1
r = r - 1
else:
while(nums[j-1] >= nums[i] and i >= 0):
i = i - 1
# 交换这两个数
nums[j-1],nums[i] = nums[i],nums[j-1]
# 对j-1之后的数组逆序排列
l = j
r = lenth-1
while(l <= r):
nums[l],nums[r] = nums[r],nums[l]
l = l + 1
r = r - 1
Tips
- 很难,没有思路,和npy讨论并看了解析之后才写出来
- 数组中可以允许有重复数字的出现,所以在判断升序的时候要<=,>=,把等于的情况忘掉了,调试了很久。
- 多练习吧。