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

最新推荐文章于 2024-10-16 00:00:43 发布

weixin_42474696

最新推荐文章于 2024-10-16 00:00:43 发布

阅读量146

点赞数

文章标签：算法 java leetcode

本文链接：https://blog.csdn.net/weixin_42474696/article/details/131251159

版权

文章介绍了两个编程题目，一个是503题，解决方法是使用取模和栈找到数组中每个元素的下一个更大元素；另一个是42题，通过栈来计算数组表示的柱状图中能接住的雨水的总面积。解题策略涉及到栈的数据结构和比较操作。

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

一、503.下一个更大元素II

重要！凡是转圈的题，都可以用取模的方式！

    public int[] nextGreaterElements(int[] nums) {
        int[] result = new int[nums.length];
        Arrays.fill(result,-1);
        LinkedList<Integer> stack = new LinkedList<>();
        stack.add(0);
        for(int i=1;i<2*nums.length;i++){
            if(nums[i%nums.length]<= nums[stack.peekLast()]){
                // stack.add(i);
                stack.add(i%nums.length);
            } else{
                while(!stack.isEmpty()&&nums[i%nums.length]> nums[stack.peekLast()]){
                    result[stack.peekLast()] = nums[i%nums.length];
                    stack.removeLast();
                }
                // stack.add(i);
                stack.add(i%nums.length);
            }
        }
        return result;
    }

11分钟

二、42. 接雨水

三个数的比较，先弹出一个中间数，然后计算高和宽，得出面积

    public int trap(int[] height) {
        int[] result = new int[height.length];
        LinkedList<Integer> stack = new LinkedList<>();
        stack.add(0);
        int sum = 0;
        for (int i = 1; i < height.length; i++) {
            if (height[i] <= height[stack.peekLast()]) {
                stack.add(i);
            } else {
                while (!stack.isEmpty() && height[i] > height[stack.peekLast()]) {
                    int mid = stack.removeLast();
                    if (!stack.isEmpty()) {
                        int h = Math.min(height[stack.peekLast()], height[i]) - height[mid];//左右高度最小值-中间柱子的高度
                        int w = i - stack.peekLast() - 1;
                        int area = h * w;
                        sum += area;
                        // stack.
                    }

                }
                stack.add(i);
            }
        }
        return sum;
    }