单调栈用来记录连续变化的数据,按照某一个顺序来记录。
每日温度
- vector<int> dailyTemperatures(vector<int>& temperatures) {
- stack<int> st;
- vector<int> answer(temperatures.size());
- st.push(0);
-
- for(int i = 1; i < temperatures.size(); i++){
- while( !st.empty() && temperatures[i] > temperatures[st.top()]){
- answer[st.top()] = i - st.top();
- st.pop();
- }
- st.push(i);
- if( temperatures[i] < temperatures[st.top()]){
- st.push(i);
- }
- }
-
- return answer;
- }
下一个最大温度I
- vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
- unordered_map<int, int> umap;
- stack<int> st;
- vector<int> result(nums1.size(), -1);
- for(int i = 0; i < nums1.size(); i++){
- umap[nums1[i]] = i;
- }
-
- for(int i = 0; i < nums2.size(); i++){
- if(!st.empty() && nums2[i] > nums2[st.top()]){
- while(!st.empty() && nums2[i] > nums2[st.top()]){
- if(umap.find(nums2[st.top()]) == umap.end()){
- st.pop();
- }
- else{
- result[umap[nums2[st.top()]]] = nums2[i];
- st.pop();
- }
- }
- st.push(i);
- }
- else{
- st.push(i);
- }
- }
- return result;
- }