文章目录
- 1. 739. 每日温度
- 2. 496.下一个更大元素 I
1. 739. 每日温度
参考文档:代码随想录
分析:
找下一个更高温度出现在几天后,即当前位置右侧出现的一个比它更大的值,如果是暴力搜索,两层for,时间复杂度O(n2), 数据范围105,时间最大是1010,这可能会超,我试了试,确实超了,用一个很大的全都一样的数组就超了。
单调栈是栈里是数组的下标,下标对应的数组元素是有序的。这里使用从栈顶到栈底递增的方法:这样遇到比栈顶更大的就可以根据下标求出距离了。遇到的元素和栈顶的元素会有三种情况:
栈顶元素大于遍历元素:入栈保留。
栈顶元素等于遍历元素:入栈保留。
栈顶元素小于遍历元素:弹出栈顶,下标相减再减一求出距离。根据栈里的单调,当前遍历元素和栈顶一直抵消,直到栈顶元素大于遍历元素。
代码:
class Solution {
public:vector<int> dailyTemperatures(vector<int>& temperatures) {vector<int> answer(temperatures.size(), 0);stack<int> st;//递增栈,存放的是下标st.push(0);for(int i = 1; i < temperatures.size(); i++){if(temperatures[i] < temperatures[st.top()]){//小于栈顶st.push(i);}else if(temperatures[i] == temperatures[st.top()]){st.push(i);}else{while(!st.empty() && temperatures[i] > temperatures[st.top()]){answer[st.top()] = i - st.top();st.pop();}st.push(i);} }if(!st.empty()){answer[st.top()] = 0;st.pop();}return answer;}
};
2. 496.下一个更大元素 I
参考文档:代码随想录
分析:
和温度一样要找出右侧第一个比当前元素大的值,但是在找的时候需要查询当前元素在nums1中在不在,在的话保存值,而不是距离。因为用到查询是不是在nums1中存在,unordered_map<int, int>哈希表的底层逻辑查找效率是O(1),保存nums1中的数组下标和数组元素。
代码:
class Solution {
public:vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {stack<int> st;vector<int> answer(nums1.size(), -1);if(nums1.size() == 0) return answer;unordered_map<int, int> umap;for(int i = 0; i < nums1.size(); i++){umap[nums1[i]] = i;//保存值和下标,下标用来更新result}st.push(0);for(int i = 1; i < nums2.size(); i++){if(nums2[i] < nums2[st.top()]){st.push(i);}else if(nums2[i] == nums2[st.top()]){st.push(i);}else{while(!st.empty() && nums2[i] > nums2[st.top()]){if(umap.count(nums2[st.top()]) > 0) answer[umap[nums2[st.top()]]] = nums2[i];st.pop();}st.push(i);}}return answer;}
};