Leetcode算法——31、下一个序列

给定一个序列，按照字典序，输出下一个序列，使得到的新数组的字典序恰好大于原数组。

如果不存在这样的排列，就将原数组从小到大排序。

替换必须就地进行，不要分配额外的内存。

示例：
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)