【Hot100】LeetCode—42. 接雨水

山脚ice

于 2024-08-12 13:22:36 发布

阅读量157

点赞数 3

分类专栏： # Hot100 文章标签： leetcode java 算法

本文链接：https://blog.csdn.net/weixin_44382896/article/details/141128918

版权

Hot100 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

1- 思路

单调栈

目标——> 求右侧第一个比当前元素大的元素 ——> 单调递增栈

2- 实现

⭐接雨水——题解思路

在这里插入图片描述

class Solution {
    public int trap(int[] height) {
        if(height.length == 0){
            return 0;
        }

        int sum = 0;
        Stack<Integer> st = new Stack<>();
        st.push(0);
        for(int i = 1 ; i < height.length;i++){
            if(height[i] <= height[st.peek()]){
                st.push(i);
            }else{
                while(!st.isEmpty() && height[i] > height[st.peek()]){
                    int right = i;
                    int mid = st.peek();
                    st.pop();
                    if(!st.isEmpty()){
                        int left = st.peek();
                        int h = Math.min(height[left],height[right])-height[mid];
                        int width = right-left-1;
                        int nowArea = h*width;
                        sum+=nowArea;
                    }
                }
                st.push(i);
            }
        }
        return sum;
    }
}

3- ACM 实现

public class trap {

    public static int trap(int[] nums){
        Stack<Integer> st = new Stack<>();
        int res = 0;
        // 遍历
        st.push(0);
        for(int i = 1 ; i <nums.length;i++){
            if(nums[i]<=nums[st.peek()]){
                st.push(i);
            }else{
                while (!st.isEmpty() && nums[i] > nums[st.peek()]){
                    int right = i;
                    int mid = st.peek();
                    st.pop();
                    if(!st.isEmpty()){
                        int left = st.peek();
                        int h = Math.min(nums[left],nums[right])-nums[mid];
                        int width = right-left-1;
                        int nowArea = width*h;
                        res+=nowArea;
                    }
                }
                st.push(i);
            }
        }
        return res;
    }

    
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        System.out.println("输入数组长度");
        int n = sc.nextInt();
        int[] nums = new int[n];
        for(int i = 0 ; i < n ;i++){
            nums[i] = sc.nextInt();
        }
        System.out.println("结果是"+trap(nums));
    }
}