Day50- 单调栈part01

一、每日温度

题目一:739. 每日温度

739. 每日温度

给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替

如果明天的温度比今天高,那么栈顶的元素就可以出栈。栈内保存的是索引,栈顶的元素是当前未被高温度覆盖的最近的一天。

class Solution {
public:
    vector dailyTemperatures(vector& temperatures) {
        int n = temperatures.size();
        vector answer(n);
        stack s;  
        for(int i = 0; i < n; ++i){
            while(!s.empty() && temperatures[s.top()] < temperatures[i]){
                int preIndex = s.top();
                answer[preIndex] = i - preIndex; 
                s.pop();
            }
            s.push(i); 
        }

        while(!s.empty()){
            answer[s.top()] = 0;
            s.pop();
        }
        return answer;
    }
};

二、下一个更大元素 I

题目一:496. 下一个更大元素 I

496. 下一个更大元素 I

nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。

给你两个 没有重复元素 的数组 nums1 和 nums2 ,下标从 0 开始计数,其中nums1 是 nums2 的子集。

对于每个 0 <= i < nums1.length ,找出满足 nums1[i] == nums2[j] 的下标 j ,并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素,那么本次查询的答案是 -1 。

返回一个长度为 nums1.length 的数组 ans 作为答案,满足 ans[i] 是如上所述的 下一个更大元素 。

创建一个映射用于存储nums2中每个数的下一个更大元素,然后再用创建的映射查找nums1中元素的下一个更大元素。

class Solution {
public:
    vector nextGreaterElement(vector& nums1, vector& nums2) {
        unordered_map numToNextGreater;
        stack nextGreaterStack;
        for(int num : nums2){
            while(!nextGreaterStack.empty() && nextGreaterStack.top() < num){
                numToNextGreater[nextGreaterStack.top()] = num;
                nextGreaterStack.pop();
            }
            nextGreaterStack.push(num);
        }
        while(!nextGreaterStack.empty()){
            numToNextGreater[nextGreaterStack.top()] = -1;
            nextGreaterStack.pop();
        }
        vector answer(nums1.size());
        for(int i = 0; i < nums1.size(); ++i){
            answer[i] = numToNextGreater[nums1[i]];
        }
        return answer;
    }
};

 

你可能感兴趣的:(代码随想录算法训练营,算法,数据结构,leetcode)