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

503.下一个更大元素II

题目链接
文章讲解

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

42. 接雨水

题目链接
文章讲解

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