力扣42. 接雨水(单调栈）

最新推荐文章于 2024-04-29 09:11:57 发布

SEVENY_

最新推荐文章于 2024-04-29 09:11:57 发布

阅读量285

点赞数

分类专栏：单调栈工作机试文章标签： leetcode c语言算法单调栈

本文链接：https://blog.csdn.net/SEVENY_/article/details/125628099

版权

工作机试同时被 2 个专栏收录

7 篇文章 0 订阅

订阅专栏

单调栈

4 篇文章 0 订阅

订阅专栏

42. 接雨水

int trap(int* height, int heightSize){
    int sstack[heightSize];
    int top = 0;
    int height_min;
    int bound_height;
    int cnt = 0;
    for (int i = 0; i < heightSize; i++) {
        while((top > 0) && (height[sstack[top - 1]] < height[i])) {
            height_min = height[sstack[top - 1]];
            top--;
            if (top <= 0) {
                break;
            }
            if (height[sstack[top - 1]] > height[i]) {
                bound_height = height[i];
            } else {
                bound_height = height[sstack[top - 1]];
            }
            cnt = cnt + (bound_height - height_min) * (i - sstack[top - 1] - 1);
        }
        sstack[top] = i;
        top++;
    }
    return cnt;
}