[LeetCode] 556. Next Greater Element III

题目链接: https://leetcode.com/problems/next-greater-element-iii/description/

Description

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 1:

Input: 12
Output: 21

Example 2:

Input: 21
Output: -1

解题思路

  1. 将整数拆分成一个个数字,从后往前寻找第一个比其后面数字小的位置,记为keyPos
  2. 寻找keyPos后面数字中,比其大的最小的数字,与之交换;
  3. keyPos后面数字从小到大排序。

例子: 1346532

  1. 从后往前找 13**4**6532,找到keyPos为下标2,即数字4;
  2. 将数字4与其后面的数字 6532 中比它大的最小的数字 5 交换,得到 13**5**6432;
  3. keyPos后面的数字,即 6432 从小到大排序得到结果,1352346

Code

class Solution {
public:
    int nextGreaterElement(int n) {
        vector<int> nums;
        int temp = n;
        long res = 0;

        while (temp != 0) {
            nums.push_back(temp % 10);
            temp /= 10;
        }

        int keyPos = 1;
        while (keyPos < nums.size() && nums[keyPos] >= nums[keyPos - 1]) keyPos++;
        if (keyPos >= nums.size()) return -1;        

        int firstBigPos = keyPos - 1;
        for (int i = firstBigPos - 1; i >= 0; --i) {
            if (nums[i] > nums[keyPos] && nums[i] < nums[firstBigPos]) firstBigPos = i;
        }

        temp = nums[keyPos];
        nums[keyPos] = nums[firstBigPos];
        nums[firstBigPos] = temp;

        sort(nums.begin(), nums.begin() + keyPos, [](int a, int b) {return a > b;});

        for (int i = nums.size() - 1; i >= 0; --i) res = res * 10 + nums[i];

        return res > INT_MAX ? -1 : res;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值