leetcode 31 下一个排列
z

实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。

如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。

必须原地修改,只允许使用额外常数空间。

以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1


  • 从后往前遍历,每一次判断nums[i]是否大于nums[i+1],如果大于就继续,不大于就暂时停止。
  • 遍历一遍得到i之后,如果i==-1,说明整个数组都是递减的,此时的下一个数值为整个数组的reverse
  • 如果i在中间的某一个idx,从i往后遍历,找到最小的那个大于nums[i]的数,swap这两个数
  • reverse之后,将i后面的整个数组reverse

举个例子:

1
2
3
4
5
6
7
        1	2	5	7	6	4	3
i j
swap:
1 2 6 7 5 4 3
i
reverse:
1 2 6 3 4 5 7
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
class Solution(object):
def nextPermutation(self, nums):
"""
:type nums: List[int]
:rtype: None Do not return anything, modify nums in-place instead.
"""
def reverse(nums, left, right):
while left < right:
nums[left], nums[right] = nums[right], nums[left]
left += 1
right -= 1

if not nums:
return
i = len(nums)-2
while i >= 0 and nums[i] >= nums[i+1]:
i -= 1
if i == -1:
nums.reverse()
return
j = i+1
while j <= len(nums)-1 and nums[j]>nums[i]:
j += 1
j -= 1
nums[i], nums[j] = nums[j], nums[i]
reverse(nums, i+1, len(nums)-1)