代码随想录算法训练营Day59 | 503.下一个更大元素II、42. 接雨水 | Python | 个人记录向

注：Day58是休息日。

503.下一个更大元素II

代码随想录：503.下一个更大元素II
Leetcode：503.下一个更大元素II

做题

和之前的739. 每日温度一样，只不过可以循环，我这边是多遍历一次nums。

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

时间复杂度: O(n)
空间复杂度: O(n)

看文章

思路一致，只是实现上有细微差别。

42. 接雨水

代码随想录：42. 接雨水
Leetcode：42. 接雨水

做题

之前面试做过这道题，处理得比较复杂，这里学学新思路。

看文章

维持一个非严格单调递减的栈，当遍历到新元素大于栈顶元素时，开始计算。雨水高度 = min(凹槽左侧高度, 凹槽右侧高度) - 凹槽底部高度，雨水宽度 = 凹槽右侧的下标 - 凹槽左侧的下标 - 1。

# 单调栈压缩版
class Solution:
    def trap(self, height: List[int]) -> int:
        stack = [0]
        result = 0
        for i in range(1, len(height)):
            while stack and height[i] > height[stack[-1]]:
                mid_height = stack.pop()
                if stack:
                    # 雨水高度是 min(凹槽左侧高度, 凹槽右侧高度) - 凹槽底部高度
                    h = min(height[stack[-1]], height[i]) - height[mid_height]
                    # 雨水宽度是 凹槽右侧的下标 - 凹槽左侧的下标 - 1
                    w = i - stack[-1] - 1
                    # 累计总雨水体积
                    result += h * w
            stack.append(i)
        return result

以往忽略的知识点小结

• 之前做 42. 接雨水 应该是用暴力求解+优化，现在要学一学单调栈的做法

个人体会

完成时间：1h20min。
心得：题相对简单，主要是学会 42. 接雨水 的单调栈做法。

​