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

目录

739 每日温度

496 下一个更大元素 |

---

739 每日温度

求后面第一个比他大的元素的位置，单调栈如果递增

求后面第一个比他小的元素的位置，单调栈需要递减

class Solution {
public:vector<int> dailyTemperatures(vector<int>& temperatures) {stack<int> st;//求的是后面第一个比他大的元素的位置，所以单调栈需要递增vector<int> res(temperatures.size(),0);//使得递增st.push(0);//存放temperatures的下标for(int i = 1;i < temperatures.size();i++){if(temperatures[i] <= temperatures[st.top()]){st.push(i);}else{while(!st.empty() && temperatures[i] > temperatures[st.top()]){//确保单调栈递增res[st.top()] = i - st.top();st.pop();}st.push(i);}}return res;}
};

时间复杂度O(n)

空间复杂度O(n) 

496 下一个更大元素 |

求后面第一个比他大的元素的位置，单调栈如果递增

求后面第一个比他小的元素的位置，单调栈需要递减

class Solution {
public:vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {stack<int>st;vector<int>res(nums1.size(),-1);unordered_map<int,int>mp;for(int i = 0;i < nums1.size();i++){//key:下标元素 value:下标mp[nums1[i]] = i;//将nums1中的数据存放到mp中}st.push(0);//存放下标for(int i = 1;i < nums2.size();i++){if(nums2[i] <= nums2[st.top()]){//求后面第一个比他大的元素的位置，单调栈如果递增st.push(i);}else{//确保单调栈单调递增while(!st.empty() && nums2[i] > nums2[st.top()]){//找到自己的位置才停止if(mp.count(nums2[st.top()]) > 0){//nums1中存在这个元素int idx = mp[nums2[st.top()]];res[idx] = nums2[i];}st.pop();}st.push(i);}}return res;}
};

时间复杂度O(m+n)//m是nums1的长度，n是nums2的长度

空间复杂度O(n)