leetcode 556. Next Greater Element III

leetcode 556. Next Greater Element III

题意:给你一个数n,让你交换这个数中的一些位,找个一个比n大的最小的数,n是32位。有的话输出那个数,没有输出-1.

思路:模拟一下交换过程就好了。

看个例子,对于230241

我们可以交换 i = 2,j = 3; i=2,j=4;    i=2;j=5使得这个数变大;很明显当i相同时,a[j]最小的数优先。

我们可以交换 i = 2,j = 4; i=3,j=4;  使得这个数变大;很明显当j相同时,i最大的数优先。

最小的话,找到最优的i,j,把[j+1,n]排个序就好了。

找到规律就可以做了。

class Solution {
public:
	int nextGreaterElement(long long  n) {
		long long a[15], l;
		stack s;
		l = 0;
		while (n)
		{
			s.push(n % 10);
			n /= 10;
		}
		while (!s.empty())
		{
			a[l++] = s.top();
			s.pop();
		}
		long long  ans = 0xffffffff;
		int lasti = -1, lastj = -1;
		for (int i = l-1; i >= 0; i--)
			for (int j = i - 1; j >= 0; j--)
			{
				if (a[i] > a[j])
				{
					if (j > lastj)
					{
						lasti = i;
						lastj = j;
					}
					else if (j == lastj )
					{
						if (a[i] < a[lasti])
						{
							lasti = i;
							lastj = j;
						}
					}
				}
			}
		if (lasti == -1 && lastj == -1) return -1;
		//cout << lasti << '\t' << lastj << '\n';
		swap(a[lasti],a[lastj]);
		sort(a+lastj+1,a+l);
		long long num = 0;
		for (int i = 0; i < l; i++)
			num = num * 10 + a[i];
		if (num > INT_MAX) return - 1;
		return num;
	}
};
有一个坑,爆int输出的-1!!!!

你可能感兴趣的:(LeetCode)