【leetcode】496. 下一个更大元素 I （单调栈）

https://leetcode-cn.com/problems/next-greater-element-i/

#include <stack>
#include <unordered_map>
using std::stack;
using std::unordered_map;

// https://leetcode-cn.com/problems/next-greater-element-i/
// 单调栈 https://blog.youkuaiyun.com/lucky52529/article/details/89155694
vector<int> StackAbout::nextGreaterElement(vector<int>& nums1, vector<int>& nums2)
{
	
	stack<int> s;
	unordered_map<int, int> ans;

	for (int i = nums2.size() - 1; i >= 0; i--)
	{
		while (!s.empty() && nums2[i] >= s.top())
		{
			s.pop(); // 栈内，最大的值永远最底下
		}	
		ans[nums2[i]] = s.empty() ? -1: s.top(); // 注意这两条语句的先后顺序
		s.push(nums2[i]); // 注意这两条语句的先后顺序
	}

	vector<int> res(nums1.size());
	// 在这之前，根本不考虑nums1这个数组
	for (int i = 0; i < nums1.size(); i++)
	{
		res[i] = ans[nums1[i]]; // 前提条件：子集
	}

	return res;
}