LeetCode刷题——单调栈

每日温度

• 题目链接
• 题意：请根据每日气温列表 temperatures ，请计算在每一天需要等几天才会有更高的温度。如果气温在这之后都不会升高，请在该位置用 0 来代替。
• 思路：单调栈，使用栈s1存储温度，栈s2存储下标，（两个栈其实可以合并为一个，存下标就可以），当栈不为空且当前温度大于栈顶温度，则ans[s2.top()]=i-s2.top()，然后pop栈顶元素。

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

下一个更大元素 I

• 题目链接
• 题意：找出 nums1 中每个元素在 nums2 中的下一个比其大的值。
• 思路：单调栈，使用栈s1存储nums2数组元素。初始化一个数组的值为-1，遍历nums2，当s1不为空循环，如果当前值大于s1.top,则t[s1.top()]=nums2[i]然后pop栈顶元素。否则退出循环,push当前元素。然后将对应元素在 nums2 中的下一个比其大的值加入答案数组。

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans(size(nums1));
        vector<int> t(10001,-1);
        stack<int> s1;
        for(int i=0;i<size(nums2);++i){
            while(!s1.empty()){
                if(nums2[i]>s1.top()){
                    t[s1.top()]=nums2[i];
                    s1.pop();
                }
                else break;
            }
            s1.push(nums2[i]);
        }
        for(int i=0;i<size(nums1);++i){
            ans[i]=t[nums1[i]];
        }
        return ans;
    }
};

下一个更大元素 II

• 题目链接
• 题意：给定一个循环数组（最后一个元素的下一个元素是数组的第一个元素），输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1。
• 思路：存在循环搜索的可能，所以遍历数组两次，s存放元素，s1存放下标，当栈不为空且当前值大于s栈顶元素，ans[s1.top()]=nums[i];然后pop栈顶元素。

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