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

一、LeetCode739. 每日温度

题目链接/视频讲解/文章讲解：https://programmercarl.com/0739.%E6%AF%8F%E6%97%A5%E6%B8%A9%E5%BA%A6.html

状态：已解决

1.思路 

       如果有同学学过单调栈的话这道题还是能看出明显解法的，因为需要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置。

单调栈的本质是空间换时间，因为在遍历的过程中需要用一个栈来记录右边第一个比当前元素高的元素，优点是整个数组只需要遍历一次。

更直白来说，就是用一个栈来记录我们遍历过的元素，因为我们遍历数组的时候，我们不知道之前都遍历了哪些元素，以至于遍历一个元素找不到是不是之前遍历过一个更小的，所以我们需要用一个容器（这里用单调栈）来记录我们遍历过的元素。

在使用单调栈的时候首先要明确如下几点：

（1）单调栈里存放的元素是什么？

单调栈里只需要存放元素的下标i就可以了，如果需要使用对应的元素，直接T[i]就可以获取。

（2）单调栈里元素是递增呢？ 还是递减呢？（以栈头到栈底的方向）

        这里我们要使用递增循序（再强调一下是指从栈头到栈底的顺序），因为只有递增的时候，栈里要加入一个元素i的时候，才知道栈顶元素在数组中右面第一个比栈顶元素大的元素是i。

即：如果求一个元素右边第一个更大元素，单调栈就是递增的，如果求一个元素右边第一个更小元素，单调栈就是递减的。

（3）使用单调栈主要有三个判断条件。

• 当前遍历的元素T[i]小于栈顶元素T[st.top()]的情况
• 当前遍历的元素T[i]等于栈顶元素T[st.top()]的情况
• 当前遍历的元素T[i]大于栈顶元素T[st.top()]的情况

        具体演示过程可以看文章讲解。

        此题由于是要找到右边的第一个比自己大的元素，故单调栈应该是递增的，且当当前元素大于栈顶时，代表找到了比前面的元素大的第一个元素，那么开始弹出栈内元素，那么弹出来的元素对应的右边比它第一大的元素的下标就为当前元素的下标，直到栈内元素不再小于当前元素时（即当前元素不再大于现在的栈顶元素），停止出栈，并把当前元素压入栈内（相当于一个单调数组找到插入位置后将元素插入进去）。

2.代码实现

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

时间复杂度：O(n)

空间复杂度：O(n)

二、496.下一个更大元素I

题目链接/视频讲解/文章讲解：https://programmercarl.com/0496.%E4%B8%8B%E4%B8%80%E4%B8%AA%E6%9B%B4%E5%A4%A7%E5%85%83%E7%B4%A0I.html

状态：已解决

1.思路 

        这道题其实就相当于求nums2每个元素的右边比它大的第一个元素对应的值，然后根据nums1在nums2中的对应位置，把nums2每个元素对应的求解结果赋给nums1的对应元素即可。

        这里我们简化一下，在求出一个nums2的某个位置的结果时，就看nums1里有没有相等值，有的话就把这个结果直接赋给nums1的对应位置，不需要先算出nums2的结果再求nums1的对应结果。

2.代码实现

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        stack<int> s;
        vector<int> result(nums1.size(),-1);
        if(nums1.size() == 0) return result;
        unordered_map<int,int> map;
        for(int i=0;i<nums1.size();i++){
            map[nums1[i]] = i;
        }
        s.push(0);
        for(int i=1;i<nums2.size();i++){
            if(nums2[i] <= nums2[s.top()]){
                s.push(i);
            }else{
                while(!s.empty() && nums2[i]>nums2[s.top()]){
                    if(map.count(nums2[s.top()]) > 0){ //看这个nums2的值在nums1中是否出现
                        result[map[nums2[s.top()]]] = nums2[i];
                    }
                    s.pop();
                }
                s.push(i);
            }
        }
        return result;
    }
};