题目:
解释:
这道题的下一个排列指的是比当前给定数组要大的数组,举个例子,把[1,2,3]进行从小到大的全排列,如下所示
[1,2,3]
[1,3,2]
[2,1,3]
[2,3,1]
[3,1,2]
[3,2,1]
题目中[1,2,3]的下一个排列就是[1,3,2]了
思路:
弄明白题意之后,接下来要思考怎么求解下一个排列,这里我用了一种交换的思路,首先从后向前寻找第一个降序数字,例如[1,2,3]中2是第一个降序数字,那么就要寻找第一个比2大的数字,也就是3,此时交换他们的位置,并把交换之后2以及2后面的数字进行顺序排列(这个不需要排列操作,但数组变长之后就可能需要了,比如[1,3,2,4,7,6])。
代码如下所示:
class Solution:
def nextPermutation(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
if len(nums) <= 1:
return
# 先找从后向前数,第一个降序的位置
for i in range(len(nums) - 2, -1, -1):
if nums[i] < nums[i + 1]:
# 之后找到比第一个比i大的数字,交换他们的位置,交换之后的部分利用sorted函数进行排序
for k in range(len(nums) - 1, i, -1):
if nums[k] > nums[i]:
nums[i], nums[k] = nums[k], nums[i]
nums[i + 1:] = sorted(nums[i + 1:])
break
break
else:
if i == 0:
nums.sort()