LeetCode 556. 下一个更大元素 III

1. 题目

给定一个32位正整数 n，你需要找到最小的32位整数，其与 n 中存在的位数完全相同，并且其值大于n。如果不存在这样的32位整数，则返回-1。

示例 1:
输入: 12
输出: 21示例 2:
输入: 21
输出: -1

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/next-greater-element-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

从后往前找，如果找到后面比前面大的，交换两个数位
并将被交换的数位的后面从小到大排序
注意最后结果如果大于INT_MAX，返回-1

class Solution {
public:int nextGreaterElement(int number) {string str = to_string(number);//转成字符串if(str.size() <= 1)return -1;//1位数，不存在int n = str.size(), i, j;bool found = false;for(i = n-2; i >= 0; --i)//从倒数第二位开始比较，被交换数位{for(j = n-1; j > i; --j){if(str[j] > str[i]){swap(str[i],str[j]);found = true;break;}}if(found)break;}if(!found)return -1;sort(str.begin()+i+1,str.end());//交换后，i位后面的排序long num = 0;for(i = 0; i < n; ++i)num = num*10+str[i]-'0';//字符串转数字if(num > INT_MAX)return -1;return num;}
};

在这里插入图片描述

优化下，O(n) 时间复杂度

class Solution {
public:int nextGreaterElement(int number) {string str = to_string(number);if(str.size() <= 1)return -1;int n = str.size(), i=n-2, j=n-1;while(i >= 0 && str[i] >= str[i+1])i--;//从后往前找到下降点if(i == -1)return -1;while(j > i && str[j] <= str[i])j--;//找到后面比交换点大的最小数swap(str[i],str[j]);//交换i,j数位i++; j=n-1;while(i < j)swap(str[i++],str[j--]);//i后面的排序（后面是降序的）long num = 0;for(i = 0; i < n; ++i)num = num*10+str[i]-'0';if(num > INT_MAX)return -1;return num;}
};