Day58 | 739. 每日温度, 496.下一个更大元素I
每日温度
LeetCode题目:https://leetcode.cn/problems/daily-temperatures/
单调栈的简单扫盲题目。主要体现出单调栈的思想:维护一个单调递增或是单调递减的序列,以便于题目中的操作。
要维护单调栈的单调性,因此需要向栈中初始化一个元素,否则没有参照对比,没办法确定单调性。遇到小的压入栈中,相当于维护了一个单调递增的序列(从栈头到栈底的顺序)
本题使用单调栈维护后,遇到大于当前栈顶元素的值,即相当于最近的一天确定了最近一次温度比自己高的天数,然后进行栈的弹出。
代码如下:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
vector<int> result(temperatures.size(), 0);
stack<int> tempStack;
tempStack.push(0);
for (int i = 1; i < temperatures.size(); i++) {
while (!tempStack.empty() && temperatures[i] > temperatures[tempStack.top()]) {
result[tempStack.top()] = i - tempStack.top();
tempStack.pop();
}
tempStack.push(i);
}
return result;
}
};
下一个更大元素I
LeetCode题目:https://leetcode.cn/problems/next-greater-element-i/
此题需要注意的是,因为对比的序列为两个不同序列,所以在使用单调栈之前需要进行额外的操作。首先是确定,nums1是nums2的子串,所以nums1中的数值一定在nums2中出现过。因此整体上循环nums2即可。
在循环中判定nums2当前的元素是否出现在nums1中,因此可以设定一个哈希映射完成操作。此外与每日温度的操作相同。
代码如下:
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
vector<int> result(nums1.size(), -1);
unordered_map<int, int> numMap;
for (int i = 0; i < nums1.size(); i++) {
numMap[nums1[i]] = i;
}
stack<int> numStack;
numStack.push(0);
for (int i = 1; i < nums2.size(); i++) {
if (nums2[numStack.top()] > nums2[i]) {
numStack.push(i);
} else if (nums2[numStack.top()] == nums2[i]) {
numStack.push(i);
} else {
while (!numStack.empty()&& nums2[numStack.top()] < nums2[i]) {
if (numMap.count(nums2[numStack.top()])) {
int index = numMap[nums2[numStack.top()]];
result[index] = nums2[i];
}
numStack.pop();
}
numStack.push(i);
}
}
return result;
}
};