31. 下一个排列

题目

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

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

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

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

思路:<时间复杂度:O(n)>&<空间复杂度:O(1)>

从右往左遍历数组,找到第一个递减的数值 tmp 和索引 i 

 1 # [5,1,4,3,2,1,0]  

1.此前,tmp 右边已按降序排列 (即为最大值序列),下一步只需要在右边找到比 tmp 大的数值里且是最小的 索引值 j ( 等于也不行,上例中应找到数值2)

2.元素交换,再把nums[ i+1: ]排序。

class Solution:
    def nextPermutation(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        if not nums or len(nums) == 1:
            pass
        else:
            index = 0
            for i in range(len(nums)-1,0,-1):
                if nums[i] > nums[i-1]: #先找到第一个递减的数
                    index = i
                    tmp = nums[i-1] #去找一个比tmp较大的数
                    break
            if not index:
                nums.reverse()
            else:
                print(index)
                index_ano = 0
                for j in range(index-1,len(nums)-1):
                    if nums[j+1] <= tmp:
                        index_ano = j
                        break
                if index_ano:
                    nums[index-1],nums[index_ano] = nums[index_ano],nums[index-1]
                    nums[index:] = sorted(nums[index:]) 
                else: #递减且比tmp大
                    nums[index-1],nums[-1] = nums[-1],nums[index-1]
                    nums[index:] = sorted(nums[index:])
                            
原文地址:https://www.cnblogs.com/steven2020/p/10694623.html