开始我没读懂题目意思,看了下这个链接字典序算法详解。
解题思路:官方解析
class Solution:
def nextPermutation(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
nlen = len(nums)
if nlen == 1: return
left, right = 0, 0
for left in range(nlen - 2, -2, -1): # PART1:从右往左找到较小数
if left == -1: break # 都循环到这了,说明整个数组是字典序最大的排列
if nums[left] < nums[left+1]: # 为什么这样找呢?因为即使找到小的数了,也需要右边有个较大数才能交换。
break
print('left:',left)
if left >= 0: # 如果left=-1,说明整个数组都是从大到小排序的,下一个排序就是字典序最小的排序(最左最右交换得到)
for right in range(nlen - 1, left, -1): # PART2:从右往左找较大者
if nums[right] > nums[left]: # 又要比left大又不能太大
break
print('right:',right)
nums[left], nums[right] = nums[right], nums[left] # PART2.1:交换
# PART3:将交换后的后半部分从小到大排序,又因为后半部现在是从大到小排序的(在第一个循环中就是这样找到left的哦),所以只需要将最左最右互换就行了
i = left + 1
j = nlen - 1
while i < j:
nums[i], nums[j] = nums[j], nums[i]
i += 1
j -= 1
官方代码比我简洁:
class Solution:
def nextPermutation(self, nums: List[int]) -> None:
i = len(nums) - 2
while i >= 0 and nums[i] >= nums[i + 1]:
i -= 1
if i >= 0:
j = len(nums) - 1
while j >= 0 and nums[i] >= nums[j]:
j -= 1
nums[i], nums[j] = nums[j], nums[i]
left, right = i + 1, len(nums) - 1
while left < right:
nums[left], nums[right] = nums[right], nums[left]
left += 1
right -= 1