目录
739. 每日温度
496.下一个更大元素 I
739. 每日温度
题目链接:739. 每日温度
求某数右边最近的大于自己的数,采用单调栈的方法。日期早的数先进栈,每次遍历到某一天就将该天温度依次与栈中各天温度进行比较,若大于栈中温度,则该天即为栈中天的最近大于自己温度的那天。
class Solution {
public:vector<int> dailyTemperatures(vector<int>& temperatures) {vector<int> ans(temperatures.size(), 0);stack<int> st;st.push(0);for(int i = 1; i < temperatures.size(); ++i){while(!st.empty() && temperatures[st.top()] < temperatures[i]){ans[st.top()] = i - st.top();st.pop();}st.push(i);}return ans;}
};
496.下一个更大元素 I
题目链接:496. 下一个更大元素 I
最开始的思路是想只针对 nums1 中的各个数来求其下一个更大的元素,可能会存在很多重复计算;如果还是使用单调栈来实现这一题,那么其实是针对 nums2 中的每一个元素都求出其下一个更大的元素,再针对 nums1 中这些数的位置进行结果输出。
class Solution {
public:vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {vector<int> ans(nums1.size(), -1);unordered_map<int, int> umap;for(int i = 0; i < nums1.size(); ++i){umap[nums1[i]] = i;}stack<int> st;st.push(0);for(int i = 0; i < nums2.size(); ++i){while(!st.empty() && nums2[st.top()] < nums2[i]){if(umap.count(nums2[st.top()]) > 0){ans[umap[nums2[st.top()]]] = nums2[i];}st.pop();}st.push(i);}return ans;}
};