leetcode 31 下一个排列

描述:

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

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

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

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

这种题是一个个测试用例试出来的,找个时间好好思考

扩展:找出一个顺序数组的所有逆序对

class Solution:
    def nextPermutation(self, nums):
        """
        :type nums: List[int]
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        # r若果长度下于等于1,直接返回
        if len(nums) <= 1:
            return 
        
        t = -1
        for i in range(len(nums)-2, -1, -1):
            if nums[i] >= nums[i+1]:
                continue
            t = i
            break
        # 如果是逆序,sort()返回
        if t== -1:
            nums.sort()
            return 
        
        t = -1
        # 找出第一个逆序对中第一个元素所在的下标
        for i in range(len(nums)-2, -1, -1):
            if nums[i] <= nums[i+1]:
                continue
            t = i
            break
      
        tmp = nums[-1]
        idx = len(nums)-1
        while idx > t:
            if nums[idx] == tmp:
                idx = idx -1
                continue
            break
        print(idx)
        print(t)
        # 如果在t的右边存在不相等的相连序对,对换位置
        if idx != t:
            s = nums[idx]
            nums[idx] = nums[idx+1]
            nums[idx+1] = s
            return 
        # 若t的右边所有元素都相同
        else:
            tmp = t
            idx = t - 1
            # 找出第一个顺序对
            while nums[idx] >= nums[tmp]:
                idx = idx - 1
                tmp = tmp - 1
            #由于idx+1 到 len(nums)-1 之间的数是逆序关系,变成顺序关系
            self.re(nums,idx+1)
            
            if nums[idx] < nums[idx+1]:
                tmp = nums[idx]
                nums[idx] = nums[idx+1]
                nums[idx+1] = tmp
                return
            else:
                tmp = nums[idx]
                i = idx 
                while tmp>= nums[i]:
                    i = i + 1
                nums[idx] = nums[i]
                nums[i] = tmp
                return
            
    def re(self,nums, s):
        e = len(nums) -1
        while s < e:
            tmp = nums[s]
            nums[s] = nums[e]
            nums[e] = tmp
            s = s +1
            e = e -1

猜你喜欢

转载自blog.csdn.net/biubiubiu888/article/details/86494341