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

503.下一个更大元素II 

https://leetcode.com/problems/next-greater-element-ii/description/

思路： 这个问题和之前的温度问题很接近， 但是要求循环的找， 那么其实就可以把数组扩充一下， 循坏的时候i 从 0 到 2 * len(nums), 然后i 对应的 index就是 i % len(nums). 

难点： 循环起来了

class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        ans = [-1] * len(nums)
        st = deque()
        for i in range(2 * len(nums)):
            index = i % len(nums)
            while st and nums[st[-1]] < nums[index]:
                j = st.pop()
                ans[j] = nums[index]
            st.append(index)
        return ans

42. 接雨水  

思路： 这个问题很是巧妙， 之前使用双指针做过， 双指针的思路就是把面积拆分成宽度为1， 高度为左右两边最大值 - 当前高度的一个个小长方形面积。 但是单调栈的思路就不一样了， 它的拆分方式时横向拆分， 当我们要推出一个元素时， 它右边的高度是将要推入的元素， 左边的高度是推出栈顶的元素的高度， 然后取最小值计算高度差， 宽度时推入元素 和 栈顶元素下标 的差值-1。 然后更新面积就可以了。 以 [4， 2，0， 3， 2， 5] 为例， 前三个没有计算直接推入， st = [4， 2， 0] 当第三个元素被推入时， 首先被推出的0， 对的高度是 min(2, 3) - 0 =2， 宽度是1. 那先算这个小长方形面积。 然后推出2. 高度是1， 长是2. 然后直接到5，当时栈中元素对应的高度是 [4, 3, 2], 推出2， 计算小长方形面积为1， 推出3, 计算小长方形面积是 1 * 4 = 4. 

难点： 这个拆分挺巧妙的。

class Solution:
    def trap(self, height: List[int]) -> int:
        ans = 0
        st = deque()
        for i in range(len(height)):
            while st and height[st[-1]] < height[i]:
                j = st.pop()
                if st:
                    h = min(height[i] - height[j], height[st[-1]] - height[j])
                    w = i - st[-1] - 1
                    ans += h * w
            st.append(i)
        return ans