代码随想录算法训练营第62天 | 739.每日温度 496.下一个更大元素I

每日温度

如果我们单纯的遍历数组，我们不知道当前元素是否比之前的元素大，所以需要维护一个容器来记录遍历过的元素。
什么时候用单调栈？通常是一维数组，要寻找任一个元素的右边或左边第一个比自己大或小的元素的位置。时间复杂度为O(n)。
这道题的单调栈应该是递增栈还是递减栈呢（从栈头到栈底）？这道题是求右边第一个比自己大的元素的位置，小于等于栈顶元素时当前元素下标应该入栈；遇到大于栈顶元素的情况，应该记录当前元素与栈顶元素下标差，result[st.top()] = i - st.top()，并弹出栈顶元素，重复操作直到当前元素小于等于栈顶元素，此时将当前元素下标入栈。

• 情况一：当前遍历的元素T[i]大于栈顶元素T[st.top()]
• 情况二：当前遍历的元素T[i]等于栈顶元素T[st.top()]
• 情况三：当前遍历的元素T[i]小于栈顶元素T[st.top()]

由此可知，求右边第一个比自己大的元素的位置，应该维护从栈头到栈底的递增栈。

class Solution {
public:
	vector<int> dailyTemperatures(vector<int>& temperatures) {
		vector<int> result(temperatures.size(), 0);
		stack<int> st;
		st.push(0);
		for(int i = 1; i < temperatures.size(); i++) {
			while(!st.empty() && temperatures[i] > temperatures[st.top()]) {
				result[st.top()] = i - st.top();
				st.pop();
			}
			// 此时当前温度一定是小于等于栈顶元素，当前元素下标入栈
			st.push(i);
		}
		return result;
	}
};

下一个更大元素I

相对于上一道题，这道题搞了两个数组，数组1是数组2的子集，这一点有点绕，因为最终求的结果与数组1长度一致，所以我们应该构建数组1长度的结果数组，并根据题意初始化为-1。
另外要注意是找数组1中的元素在数组2中的下一个更大元素，所以单调栈的操作应该是在遍历数组2的过程中进行的，入栈出栈操作与上一题相同，只是在记录结果数组时需要判断栈顶元素是否在数组1中。
判断一个元素是否在数组中，很容易想到哈希表，建立数值到下标的映射，更方便地记录结果。

class Solution {
public:
	vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
		vector<int> result(nums1.size(), -1);
		stack<int> st;
		unordered_map<int, int> umap;
		for(int i = 0; i < nums1.size(); i++) {  // 建立数组1中数值到下标的映射
			umap[nums1[i]] = i;
		}
		for(int i = 0; i < nums2.size(); i++) {  // 遍历数组2
			while(!st.empty() && nums2[i] > nums2[st.top()]) {  // 当前元素大于栈顶元素
				if(umap.count(nums2[st.top()]) > 0) {  // 如果栈顶元素出现在数组1中，结果应该被记录
					int index = umap[nums2[st.top()]];
					result[index] = nums2[i];
				}
				st.pop();  // 不论是否结果是否记录，必须维护单调栈
			}
			// 当前元素一定小于等于栈顶元素，满足递增栈条件（从栈头到栈底）
			st.push(i);
		}
		return result;
	}
};