这个题还是有难度的,不看题解挺难想。
第一个重要的点是要找左边尽量靠右的元素和右边尽量小的元素交换,交换后对左边位置后面的元素排序/反转。这个是要自己找规律(很难找)
第二个点就是,已知上面的算法,如何找到这两个点。这个方法就更妙了。
首先要找尽量靠右的元素,尽量靠右,但要保证他右边还有比他更小的元素。所以我们要从右往左找,直至找到一个元素使得
nums[i-1] < nums[i]
这样就找到左元素。此时可以发现右边是一个降序序列,只要从右往左找,找到第一个比左元素大的元素就是右元素了(能满足尽量小的条件)
之后,由于右边是降序序列,反转即可,无需排序。
class Solution:
def nextPermutation(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
left, right = -1, -1
for i in range(len(nums)-1, 0, -1):
if nums[i-1] < nums[i]:
left = i-1
break
if left < 0:
nums.sort()
return
for i in range(len(nums)-1, left, -1):
if nums[i] > nums[left]:
right = i
break
nums[left], nums[right] = nums[right], nums[left]
# reverse
l, r = left+1, len(nums)-1
while(r-l > 0):
nums[l], nums[r] = nums[r], nums[l]
l += 1
r -= 1