LeetCode No.31 | StriveZs的博客

LeetCode No.31

LeetCode第三十一题

题目描述

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

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

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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
示例 1

输入:nums = [1,2,3]
输出:[1,3,2]
示例 2

输入:nums = [3,2,1]
输出:[1,2,3]
示例 3

输入:nums = [1,1,5]
输出:[1,5,1]
示例 4

输入:nums = [1]
输出:[1]
 

提示:

1 <= nums.length <= 100
0 <= nums[i] <= 100

代码

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.
"""
length = len(nums) # 总长度
index_a = length - 1
# 从后往前找
while index_a >= 0:
index_b = length - 1
while index_b > index_a :
# 如果大于则交换位置并对前面查找的内容进行排序
if nums[index_b] > nums[index_a]:
temp = nums[index_a]
nums[index_a] = nums[index_b]
nums[index_b] = temp
tt = nums[index_a + 1:len(nums)]
tt.sort()
nums[index_a+1:len(nums)] = tt
return nums
index_b -= 1
index_a -= 1
nums.reverse()
return nums

if __name__ == '__main__':
s = Solution()
print(s.nextPermutation(nums = [1,3,2]))
StriveZs wechat
Hobby lead  creation, technology change world.