代码随想录算法训练营第四十一天| 739. 每日温度、496.下一个更大元素 I、503.下一个更大元素II

写代码的第四十一天
开始单调栈了，以前没听过这个词，全新学习。。。

739. 每日温度

思路

根据题意可知，最后要的输出结果是找到第i天后面第一个比这一天温度高的第j天，然后输出j-i，i从0开始，代表当前数组的下标。所以根据输出我们要设置单调栈存储的是当前数组的下标，这样方便计算。分三种情况讨论，当第i天的温度小于当前栈顶元素的温度，那么将该元素入栈；当第i天的温度等于当前栈顶元素的温度，那么将该元素入栈；当第i天的温度大于当前栈顶元素的温度，那么查看当前温度的下标以及栈顶元素（栈顶元素代表的就是下标），做减法，然后将得到的数值append进result中，之后将栈顶元素弹出栈，将要插入的元素继续与新的栈顶元素比较，什么时候不大于栈顶元素了，什么时候停止，然后将要插入的元素插入栈，最后输出result。
错误第一版：result.append(i - stack[-1])这句话中没考虑清楚result存储的内容，result存储的是当前弹出的是stack栈顶元素，这个栈顶元素代表的是数组下标，所以在result中，也要找到对应的下标，在进行赋值。

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        result = []
        stack = [0]
        for i in range(len(temperatures)):
            if temperatures[i] < temperatures[stack[-1]]:
                stack.append(i)
            elif temperatures[i] == temperatures[stack[-1]]:
                stack.append(i)
            else:
                while len(stack) != 0 and temperatures[i] > temperatures[stack[-1]]:
                    result.append(i - stack[-1])
                    stack.pop()
                stack.append(i)
        return result

正确代码：

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        result = [0] * len(temperatures)
        stack = [0]
        for i in range(len(temperatures)):
            if temperatures[i] < temperatures[stack[-1]]:
                stack.append(i)
            elif temperatures[i] == temperatures[stack[-1]]:
                stack.append(i)
            else:
                while len(stack) != 0 and temperatures[i] > temperatures[stack[-1]]:
                    result[stack[-1]] = i - stack[-1]
                    stack.pop()
                stack.append(i)
        return result

496.下一个更大元素 I

思路

本题和上一题的区别在于，这是两个数组，我们需要知道两个数组之间的映射，根据nums1对nums2进行映射，找到对应的位置。所以本题的思路和上一题几乎一致。
第一步：定义输出数组result，他的长度和nums1的长度是一致的，因为我们是根据nums1去nums2找对应的，然后输出到和nums1下标位置对应的result数组中。
第二步：定义单调栈stack，栈存储的元素为数组下标。
第三步：对nums2进行单调栈操作，分三种情况讨论（这里的讨论和上一题是一样的）。1、当将要进入栈的数值小于等于栈顶元素对应的nums2时，将这个数值对应的nums2的下标入栈；2、当将要进入栈的数值大于栈顶元素对应的nums2时，将这个数值放入result中，此时这个数值对应的result下标为当前栈顶元素对应的nums1的数值的位置。
最后输出result。
错误第一版：错误原因ValueError: 3 is not in list
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
result[nums1.index(nums2[stack[-1]])] = nums2[i]
错误原因是，没考虑如果nums2有元素不在nums1中怎么办，所以应该加一个判断语句。

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        result = [-1] * len(nums1)
        stack = [0]
        for i in range(len(nums2)):
            if nums2[i] <= nums2[stack[-1]]:
                stack.append(i)
            else:
                while len(stack) != 0 and nums2[i] > nums2[stack[-1]]:
                    result[nums1.index(nums2[stack[-1]])] = nums2[i]
                    stack.pop()
                stack.append(i)
        return result

正确代码：

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        result = [-1] * len(nums1)
        stack = [0]
        for i in range(len(nums2)):
            if nums2[i] <= nums2[stack[-1]]:
                stack.append(i)
            else:
                while len(stack) != 0 and nums2[i] > nums2[stack[-1]]:
                    if nums2[stack[-1]] in nums1:
                        result[nums1.index(nums2[stack[-1]])] = nums2[i]
                    stack.pop()
                stack.append(i)
        return result

503.下一个更大元素II

思路

将nums数组和自身进行拼接！
正确代码：

class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        # 拼接一个新的 nums
        nums1 = nums.copy()
        nums.extend(nums1)       
        # 用新的 nums 大小来初始化 result
        result = [-1] * len(nums)
        if not nums:
            return result        
        # 开始单调栈
        stack = [0]
        for i in range(len(nums)):
            if nums[i] <= nums[stack[-1]]:
                stack.append(i)
            else:
                while stack and nums[i] > nums[stack[-1]]:
                    top = stack.pop()
                    result[top] = nums[i]
                stack.append(i)       
        # 最后再把结果集即 result 数组 resize 到原数组大小
        result = result[:len(nums)//2]        
        return result