从含有n个整数的无序数组中找到每个元素后面比它大的第一个数

复习时候看到的一道题，总结了总结。

题目描述

从含有n个整数的无序数组中找到每个元素后面比它大的第一个数，要求时间复杂度为O(N)。

解题思路

如果没有时间复杂度要求的话，可以使用二重遍历进行暴力解决。然而参考了各种大牛的博客，将自己的思路总结如下。

我们借助了辅助栈的方式进行解决。

对于当前遍历的第i个元素有：

1. 栈为空，则当前索引i入栈；
2. 栈不为空，如果栈顶索引元素大于等于当前元素，则将当前索引i入栈；（目的是为了保持栈从栈底到栈顶非升序）
3. 栈不为空，当前元素大于栈顶索引元素，则栈顶索引元素对应的第一个比它大的元素就是当前元素，保存当前元素；同时为了保证栈底到栈顶非升序需要将栈顶元素弹出，直到满足1、2条件。
4. 输入遍历完，栈中还剩余的元素则是不能找见之后比他大的元素。

AC代码

vector<int> findMax(const vector<int>& nums)
{
	int n = nums.size();
	if (n == 0)
		return vector<int>();
	vector<int> res(n);
	int i = 0;
	stack<int> s;
	while (i < n)
	{
        //这个if将1、2两个条件写到一起了
		if (s.empty() || nums[s.top()] >= nums[i])
			s.push(i++);
		else
		{
			res[s.top()] = nums[i];
			s.pop();
		}
	}
	//处理当前栈中剩下的元素，也就是找不到它之后比它大的元素
	while (!s.empty())
	{
		res[s.top()] = INT_MAX;
		s.pop();
	}
	return res;
}