(Python)LeetCode 31:下一个排列

题目:

思路:

这道题在最后成功之前,我总共错了三次,都是一些情况没有考虑到,最后整理思路如下:

首先,我们应该倒序去看这个列表会比较好理解,从后往前,一旦某个下标的元素比前一个元素大,那么肯定将这两个元素调换即可,但是,由于这里要求的是比原本的数大的最小值,所以,我们应该要尽可能的把小的数放在前面,大的数放在后面,而不能是简单的替换,比如:[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的解题方法欢迎大家一起讨论~

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值