算法刷题Day 58 每日温度+下一个更大元素I

最新推荐文章于 2024-07-17 23:24:38 发布

benobug

最新推荐文章于 2024-07-17 23:24:38 发布

阅读量556

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/benobug/article/details/132075705

版权

Day 58 单调栈

739. 每日温度

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        vector<int> rst(temperatures.size());
        vector<int> decsStk; // 单调递减栈

        for (int i = 0; i < temperatures.size(); i++)
        {
            while (!decsStk.empty() && temperatures[decsStk.back()] < temperatures[i])
            {
                rst[decsStk.back()] = i - decsStk.back();
                decsStk.pop_back();
            }

            decsStk.push_back(i);
        }

        return rst;
    }
};

496. 下一个更大元素I

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> decsStk; // 存放nums2的下标
        unordered_map<int, int> table;
        
        for (int i = 0; i < nums2.size(); ++i)
        {
            while (!decsStk.empty() && nums2[decsStk.back()] < nums2[i])
            {
                table[nums2[decsStk.back()]] = nums2[i];
                decsStk.pop_back();
            }
            decsStk.push_back(i);
        }

        vector<int> rst(nums1.size(), -1);
        for (int i = 0; i < nums1.size(); i++)
        {
            if (table.find(nums1[i]) != table.end())
            {
                rst[i] = table[nums1[i]];
            }
        }

        return rst;
    }
};