LeetCode 496. 下一个更大元素 I 题解 C/C++

方法一：暴力
对于每一个nums1，先在nums2中找到它，然后向右遍历找到第一个大于nums1[i]的元素.
方法二：单调栈
根据题意，数组 nums1 视为询问。我们可以：
先对 nums2 中的每一个元素，求出它的右边第一个更大的元素；将上一步的对应关系放入哈希表（HashMap）中；再遍历数组 nums1，根据哈希表找出答案。
具体步骤就是：当我们遇到一个新的元素 nums2[i] 时，我们判断栈顶元素是否小于 nums2[i]，如果是，那么栈顶元素的下一个更大元素即为 nums2[i]，我们将栈顶元素出栈。重复这一操作，直到栈为空或者栈顶元素大于 nums2[i]。此时我们将 nums2[i] 入栈，保持栈的单调性，并对接下来的 nums2[i + 1], nums2[i + 2] … 执行同样的操作。

参考链接

class Solution1 {
public:
	vector<int> nextGreaterElement(vector<int> &nums1,vector<int> &nums2) {
		vector<int> vec;
		for(vector<int>::iterator it = nums1.begin();it!=nums1.end();it++) {
			int flag = 0;
			for(vector<int>::iterator cur = nums2.begin();cur!=nums2.end();cur++) {
				if(*it==*cur) {
					cur++;
					while(cur!=nums2.end()) {
						if(*cur>*it){
							vec.push_back(*cur);
							flag = 1;
							break;
						}
						cur++;
					}
				}
				if(flag||cur==nums2.end())break;
			}
			if(flag==0)vec.push_back(-1);
		}
		return vec;
	}

};

/*
方法二：单调栈
*/
class Solution {
public:
	vector<int> nextGreaterElement(vector<int>& nums1,vector<int> &nums2) {
		int len1 = nums1.size();
		int len2 = nums2.size();
		stack<int> mono_stk;
		vector<int> vec;
		map<int,int> hash_table;
		mono_stk.push(nums2[0]);
		for(int i= 1;i<len2;i++) {
			if(mono_stk.top()<nums2[i]) {
				while(!mono_stk.empty()&&mono_stk.top()<nums2[i]) {
					int key = mono_stk.top();
					mono_stk.pop();
					hash_table.insert(pair<int,int>(key,nums2[i]));
				}
			}
			mono_stk.push(nums2[i]);
		}
		while(!mono_stk.empty()) {
			hash_table.insert(pair<int,int>(mono_stk.top(),-1));
			mono_stk.pop();
		}
		for(int i = 0;i<len1;i++) {
			vec.push_back(hash_table[nums1[i]]);
		}
		return vec;
	}
};