Day58 | 739. 每日温度, 496.下一个更大元素I

Day58 | 739. 每日温度, 496.下一个更大元素I

每日温度

LeetCode题目:https://leetcode.cn/problems/daily-temperatures/

  单调栈的简单扫盲题目。主要体现出单调栈的思想:维护一个单调递增或是单调递减的序列,以便于题目中的操作。

  要维护单调栈的单调性,因此需要向栈中初始化一个元素,否则没有参照对比,没办法确定单调性。遇到小的压入栈中,相当于维护了一个单调递增的序列(从栈头到栈底的顺序)

  本题使用单调栈维护后,遇到大于当前栈顶元素的值,即相当于最近的一天确定了最近一次温度比自己高的天数,然后进行栈的弹出。

  代码如下:

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

下一个更大元素I

LeetCode题目:https://leetcode.cn/problems/next-greater-element-i/

  此题需要注意的是,因为对比的序列为两个不同序列,所以在使用单调栈之前需要进行额外的操作。首先是确定,nums1是nums2的子串,所以nums1中的数值一定在nums2中出现过。因此整体上循环nums2即可。

  在循环中判定nums2当前的元素是否出现在nums1中,因此可以设定一个哈希映射完成操作。此外与每日温度的操作相同。

  代码如下:

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> result(nums1.size(), -1);

        unordered_map<int, int> numMap;
        for (int i = 0; i < nums1.size(); i++) {
            numMap[nums1[i]] = i;
        }
        stack<int> numStack;
        numStack.push(0);
        for (int i = 1; i < nums2.size(); i++) {
            if (nums2[numStack.top()] > nums2[i]) {
                numStack.push(i);
            } else if (nums2[numStack.top()] == nums2[i]) {
                numStack.push(i);
            } else {
                while (!numStack.empty()&& nums2[numStack.top()] < nums2[i]) {
                    if (numMap.count(nums2[numStack.top()])) {
                        int index = numMap[nums2[numStack.top()]];
                        result[index] = nums2[i];
                    }
                    numStack.pop();    
                }
                numStack.push(i);
            }
        }

        return result;
    }
};

你可能感兴趣的:(数据结构,算法,leetcode)