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

最新推荐文章于 2024-10-31 19:07:52 发布

冷酷的橙子

最新推荐文章于 2024-10-31 19:07:52 发布

阅读量242

点赞数

分类专栏：代码随想录算法训练营文章标签：算法 leetcode 数据结构 Powered by 金山文档

本文链接：https://blog.csdn.net/weixin_44845049/article/details/129567286

版权

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

49 篇文章 0 订阅

订阅专栏

文章提供了两个C++类Solution，分别解决LeetCode中的两道题目。一个是找到给定循环数组中每个元素的下一个更大元素，使用了栈数据结构；另一个是计算给定数组中可以接住的雨水量，利用动态规划计算最大高度差。

摘要由CSDN通过智能技术生成

● 503.下一个更大元素II

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

● 42. 接雨水

class Solution {
public:
    int trap(vector<int>& height) {
        vector<vector<int>> dp(height.size(), vector<int>(2, -1));
        dp[0][0] = -1;
        dp[height.size() - 1][1] = -1;
        for (int i = 1; i < height.size(); i ++) {
            dp[i][0] = max(height[i - 1], dp[i - 1][0]);
        }
        for (int i = height.size() - 2; i >= 0; i --) {
            dp[i][1] = max(height[i + 1], dp[i + 1][1]);
        }
        int res = 0;
        for (int i = 0; i < height.size(); i ++) {
            //printf("%d %d %d\n", dp[i][0], height[i], dp[i][1]);
            if (dp[i][0] > height[i] && height[i] < dp[i][1]) {
                res += min(dp[i][0], dp[i][1]) - height[i];
            }
        }
        return res;
    }
};