Leetcode 单调栈问题总结(超详细!!!)

0x00

单调栈主要回答这样的几种问题

  • 比当前元素更大的下一个元素
  • 比当前元素更大的前一个元素
  • 比当前元素更小的下一个元素
  • 比当前元素更小的前一个元素

0x01 问题一

维护一个单调递减的栈。

Leetcode 42:接雨水(超详细的解法!!!)

Leetcode 496:下一个更大元素 I(超详细的解法!!!)

Leetcode 503:下一个更大元素 II(超详细的解法!!!)

Leetcode 739:每日温度(超详细的解法!!!)

class Solution:
    def nextGreaterElement(self, nums):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        stack = list()
        res = [-1]*len(nums)
        for i, n in enumerate(nums):
            while stack and nums[stack[-1]] < n:
                res[stack.pop()] = n
            stack.append(i)
        return res

0x02 问题二

维护一个单调递减的栈。

Leetcode 901:股票价格跨度(超详细的解法!!!)

Leetcode 239:滑动窗口最大值(超详细的解法!!!)(更明确为区间最大元素问题)

class Solution:
    def preGreaterElement(self, nums):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        stack = list()
        res = [-1]*len(nums)
        for i, n in enumerate(nums):
            while stack and nums[stack[-1]] < n:
                stack.pop()
            if stack:
                res[i] = nums[stack[-1]]
            stack.append(i)
        return res

0x03 问题三

维护一个单调递增的栈。

Leetcode 84:柱状图中最大的矩形(超详细的解法!!!)

class Solution:
    def nextSmallerElement(self, nums):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        stack = list()
        res = [-1]*len(nums)
        for i, n in enumerate(nums):
            while stack and nums[stack[-1]] > n:
                res[stack.pop()] = n
            stack.append(i)
        return res

0x04 问题四

维护一个单调递增的栈。

class Solution:
    def preSmallerElement(self, nums):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        stack = list()
        res = [-1]*len(nums)
        for i, n in enumerate(nums):
            while stack and nums[stack[-1]] > n:
                stack.pop()
            if stack:
                res[i] = nums[stack[-1]]
            stack.append(i)
        return res

至于最后一点要说的就是,如何确定是使用严格单调栈还是非严格单调栈只要根据题意确定我们栈中是否可以存放相同元素即可。

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题,希望大家指出!!!

你可能感兴趣的:(Data,Structures,and,Algorithms,单调栈)