题目:
思路:
这道题在最后成功之前,我总共错了三次,都是一些情况没有考虑到,最后整理思路如下:
首先,我们应该倒序去看这个列表会比较好理解,从后往前,一旦某个下标的元素比前一个元素大,那么肯定将这两个元素调换即可,但是,由于这里要求的是比原本的数大的最小值,所以,我们应该要尽可能的把小的数放在前面,大的数放在后面,而不能是简单的替换,比如:[3,1,4,2],正确的替换结果应该是[3,2,1,4],可以看到,并不是简单的把两个相邻元素(1和4)进行替换,而是把1后面的数,即:2和4中挑选比1大的最小值,也就是2,替换到当前1的位置,(代码里我用val记录了需要被替换的值,相当于是这里的1),然后剩下的已经被遍历过的后几位数(这里我放在了新的列表new里面),则是按照排列的顺序添加到原来的nums列表里即可。
另外还要注意的是,可能列表的元素有重复的情况,可能正好是val值是一个重复的值,我们要找到比它大的值,但是只用new,index()返回的是第一个找到的下标,也就是最小的下标,我们要的是最大的,所以这里用了一个while循环,来找到最大的下标,继而找到比他大的最小值。
并且还要考虑到如果nums列表本身就是按照从大到小的顺序的,那么直接用sort,将列表从小到大排序即可。
代码:
# -- coding: utf-8 --
# @Time : 2021/12/2 9:57
# @Author : 木可木可
# @File : 下一个排列.py
class Solution:
def nextPermutation(self, nums) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
count = 0
new = [] # 用于存储新数据的列表
for i in nums: # 统计列表长度
count += 1
i = count - 1
while i > 0:
if nums[i] > nums[i - 1]: # 判断是否要开始更新数组
val = nums[i-1] # 记录当前下标的前一个数据
new.append(nums[i-1]) # 将元素添加到新列表
new.append(nums[i])
new = sorted(new) # 对新的列表排序,用于对nums后面的数据重新排序
index = new.index(val)
while new[index+1] == val: # 避免有相同元素的情况
index = index + 1
nums[i-1] = new[index+1] # 将当前下标的前一个数据值替换为 new中大于它的最小值
del new[index + 1] # 删除掉new中已经被使用过的那个数
break
else:
new.append(nums[i])
i -= 1
if i == 0: # 列表从大到小排列,直接用sort进行从小到大的排序
nums.sort(reverse=False)
return nums
while i < count:
nums[i] = min(new)
del new[0] # 把new中当前使用过的元素删掉,也就是第一个元素
i += 1
return nums
if __name__ == '__main__':
s = Solution()
# nums = [1,2,3] # 测试用例
# nums = [3,2,1]
# nums = [1]
# nums = [1,2,4,3]
# nums = [1,5,3,4,2,1]
# nums = [1,5,1]
nums = s.nextPermutation(nums)
print(nums)
结果:
关于LeetCode的解题方法欢迎大家一起讨论~