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

最新推荐文章于 2024-05-21 23:16:53 发布

冰冰的coco

最新推荐文章于 2024-05-21 23:16:53 发布

阅读量196

点赞数

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

本文链接：https://blog.csdn.net/qq_44144025/article/details/127914585

版权

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

52 篇文章 0 订阅

订阅专栏

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;
    }
};