接雨水算法

大山峰石

已于 2025-03-11 12:22:43 修改

阅读量102

点赞数

文章标签：算法数据结构 c++

于 2023-05-04 15:00:16 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_57722445/article/details/130360457

版权

在这里插入图片描述

int trap(vector<int>& height) {
        int ans = 0;
        stack<int> stk;
        int n = height.size();
        for (int i = 0; i < n; ++i) {
            while (!stk.empty() && height[i] > height[stk.top()]) {
                int top = stk.top();
                stk.pop();
                if (stk.empty()) {
                    break;
                }
                int left = stk.top();
                int currWidth = i - left - 1;
                int currHeight = min(height[left], height[i]) - height[top];
                ans += currWidth * currHeight;
            }
            stk.push(i);
        }
        return ans;
    }