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

• 今日学习的文章链接，或者视频链接

第十章 单调栈part02

• 自己看到题目的第一想法

• 看完代码随想录之后的想法

503:

注意对环形数组的处理：

class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        result = [-1] * len(nums)
        stack = []
        for i in range(len(nums)*2):
            new_i = i%len(nums)
            while(len(stack) != 0 and nums[new_i] > nums[stack[-1]]):
                    result[stack[-1]] = nums[new_i]
                    stack.pop()
            stack.append(new_i)
        return result

42:

 

class Solution:
    def trap(self, height: List[int]) -> int:
        # stack储存index，用于计算对应的柱子高度
        stack = [0]
        result = 0
        for i in range(1, len(height)):
            # 情况一
            if height[i] < height[stack[-1]]:
                stack.append(i)

            # 情况二
            # 当当前柱子高度和栈顶一致时，左边的一个是不可能存放雨水的，所以保留右侧新柱子
            # 需要使用最右边的柱子来计算宽度
            elif height[i] == height[stack[-1]]:
                stack.pop()
                stack.append(i)

            # 情况三
            else:
                # 抛出所有较低的柱子
                while stack and height[i] > height[stack[-1]]:
                    # 栈顶就是中间的柱子：储水槽，就是凹槽的地步
                    mid_height = height[stack[-1]]
                    stack.pop()
                    if stack:
                        right_height = height[i]
                        left_height = height[stack[-1]]
                        # 两侧的较矮一方的高度 - 凹槽底部高度
                        h = min(right_height, left_height) - mid_height
                        # 凹槽右侧下标 - 凹槽左侧下标 - 1: 只求中间宽度
                        w = i - stack[-1] - 1
                        # 体积：高乘宽
                        result += h * w
                stack.append(i)
        return result

• 自己实现过程中遇到哪些困难

• 今日收获，记录一下自己的学习时长