版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/HappyRocking/article/details/83662144
给定一个序列,按照字典序,输出下一个序列,使得到的新数组的字典序恰好大于原数组。
如果不存在这样的排列,就将原数组从小到大排序。
替换必须就地进行,不要分配额外的内存。
示例:
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
思路
可以使用字典序算法,详见字典序算法详解。
python实现
def nextPermutation(nums):
"""
:type nums: List[int]
:rtype: void Do not return anything, modify nums in-place instead.
寻找基于字典序的下一个排列。
"""
l = len(nums)
# 从右向左查询第一个小于右邻的元素
for i in range(l-2, -1, -1):
if nums[i+1] > nums[i]:
break
else: # 没有找到,说明为降序排列
nums[:] = nums[::-1]
return
# 从右向左查询第一个大于nums[i]的元素
for j in range(l-1, -1, -1):
if nums[j] > nums[i]:
break
# 交换i和j
nums[i], nums[j] = nums[j], nums[i]
# i后面的序列进行反转
nums[i+1:l] = nums[-1:i:-1]
if '__main__' == __name__:
nums = [3,2,1]
nextPermutation(nums)
print(nums)