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

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

503.下一个更大元素II

// 这个循环遍历真的很重要，就是用%去实现。
class Solution {
    public int[] nextGreaterElements(int[] nums) {
        // 这个在后边拼接真没想到
        if (nums.length == 1) {
            return new int[]{-1};
        }
        int len = nums.length;
        int[] result = new int[len];
        Stack<Integer> stack = new Stack<>();
        for (int i = 0; i < len; i++) {
            result[i] = -1;
        }
        // 循环就是在拼接一个相同的数组在后边，也可以用先乘2然后再去%原来len的方法。
        // 都是思想啊。
        for (int i = 0; i < len*2; i++) {
            while (!stack.isEmpty() && nums[i%len] > nums[stack.peek()]) {
                result[stack.peek()] = nums[i%len];
                stack.pop(); 
            }
            stack.push(i%len);
        }
        return result;
    }
}

42. 接雨水

class Solution {
    public int trap(int[] height) {
        // 这个题经常被当作大厂面试题
        // 双指针法，
        // 可以通过行也可以通过列，
        // 现在的方法是通过列去求
        int len = height.length;
        // 就是目标列的左右最高
        int[] leftHeight = new int[len];
        int[] rightHeight = new int[len];

        leftHeight[0] = height[0];
        // 从左开始是有原因的
        for (int i = 1; i < len; i++) {
            leftHeight[i] = Math.max(height[i], leftHeight[i-1]);
        }
        rightHeight[len - 1] = height[len-1];
        // 从右开始是有原因的
        for (int i = len-2; i >= 0; i--) {
            rightHeight[i] = Math.max(height[i], rightHeight[i+1]);
        }

        int sum = 0;
        for (int i = 0; i < len-1; i++) {
            // 最后找到最矮的一边，然后去减柱子的高度。
            int count = Math.min(rightHeight[i],leftHeight[i]) - height[i];
            if (count > 0) {
                sum += count;
            }
        }
        return sum;
    }
}


class Solution {
// 就是从水槽底部一层一层遍布，
    public int trap(int[] height) {
        // 这个题经常被当作大厂面试题
        // 单调栈法
        // 可以通过行也可以通过列，
        // 现在的方法是通过列去求
        // 关于单调栈的顺序给大家一个总结： 739. 每日温度 (opens new window)中求一个元素右边第一个更大元素，单调栈就是递增的，84.柱状图中最大的矩形 (opens new window)求一个元素右边第一个更小元素，单调栈就是递减的。
        int len = height.length;
        // 栈中存的是下标
        Stack<Integer> stack = new Stack<>();
        stack.push(0);

        int sum = 0;

        for (int i = 1; i < len; i++) {
            if (height[i] < height[stack.peek()]) {
                stack.push(i);
            } else if (height[i] == height[stack.peek()]) {
                stack.pop();
                stack.push(i);
            } else {
                while (!stack.isEmpty() && height[i] > height[stack.peek()]) {
                    int num = stack.peek();
                    stack.pop();
                    // 不为空才可以处理接下来的逻辑
                    if (!stack.isEmpty()) {
                        // 高度
                        int h = Math.min(height[stack.peek()],height[i]) - height[num];
                        // 宽度要记得减一
                        int m = i - stack.peek() -1;
                        sum += h*m;
                    }
                }
                stack.push(i);
            }
        }
        return sum;
        
    }
}