代码随想录算法训练营第五十九天| LeetCode503.下一个更大元素II 42. 接雨水

冰冰的coco

于 2022-11-18 00:46:30 发布

阅读量325

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/qq_44144025/article/details/127914585

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

52 篇文章

订阅专栏

本文解析了LeetCode上的两道经典栈算法题目：503. 下一个更大元素 II 和 42. 接雨水。通过详细阐述解题思路及代码实现，帮助读者理解如何利用栈来高效解决这类问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

503.下一个更大元素II

题目：力扣

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

42. 接雨水

题目：力扣

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

总结

题型：