Leetcode-556. 下一个更大元素 III -python

最新推荐文章于 2021-08-17 09:36:56 发布

简米

最新推荐文章于 2021-08-17 09:36:56 发布

阅读量273

点赞数

分类专栏： Leetcode 文章标签： leetcode python 算法

本文链接：https://blog.csdn.net/JamieLuo/article/details/108212467

版权

Leetcode 专栏收录该内容

83 篇文章 1 订阅

订阅专栏

给定32位正整数n，寻找位数相同且值大于n的最小32位整数。例如，输入230241，输出230412。思路是从末位开始找到第一个降序位置，反转该位置后的数字并交换此位置与其后首个大于它的数字。

摘要由CSDN通过智能技术生成

题目

给定一个32位正整数 n，你需要找到最小的32位整数，其与 n 中存在的位数完全相同，并且其值大于n。如果不存在这样的32位整数，则返回-1。
链接：https://leetcode-cn.com/problems/next-greater-element-iii/

Given a positive 32-bit integer n, you need to find the smallest 32-bit integer which has exactly the same digits existing in the integer n and is greater in value than n. If no such positive 32-bit integer exists, you need to return -1.

Example:

Input: 230241
Output: 230412

思路及代码

找到第一位开始降序的位置：比如230241-从末位1开始14是升序，到2就降序了
reverse这一位之后的所有，得到230214
交换这一位与这一位之后第一个大于他的数字，即交换2和4得到230412

class Solution:
    def nextGreaterElement(self, n: int) -> int:
        s = list(map(int,str(n)))
        i = len(s) - 1
        # 找到第一位开始降序的位置：如230241中的4的位置
        while i >= 1 and s[i] <= s[i-1]:
            i -= 1
        if i == 0:
            return -1
        s[i:len(s)] = reversed(s[i:len(s)])
        for j in range(i,len(s)):
            if s[j] > s[i-1]:
                break
        s[i-1], s[j] = s[j], s[i-1]
        ans = int("".join(map(str,s)))
        if ans > 2**31-1: # (1<<31)-1
            return -1
        return ans