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

冰冰的coco

已于 2022-11-18 00:20:26 修改

阅读量203

点赞数

分类专栏：代码随想录算法训练营文章标签： leetcode 算法 c++

于 2022-11-18 00:20:04 首次发布

本文链接：https://blog.csdn.net/qq_44144025/article/details/127914451

版权

代码随想录算法训练营专栏收录该内容

52 篇文章 0 订阅

订阅专栏

739. 每日温度

题目：力扣

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

496.下一个更大元素 I

题目：力扣

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int,int> um;
        for(int i = 0; i < nums1.size(); ++i){
            um[nums1[i]] = i;
        }
        vector<int> ans(nums1.size(), -1);
        stack<int> s;
        s.push(0);
        for(int i = 1; i < nums2.size(); ++i){
            while(!s.empty() && nums2[i] > nums2[s.top()]){
                if(um.find(nums2[s.top()]) != um.end()){
                    ans[um[nums2[s.top()]]] = nums2[i];
                }
                
                s.pop();
            }
            s.push(i);
        }
        return ans;
    }
};