代码随想录算法训练营19期第58天

DWcsdnNET

已于 2023-09-21 03:55:09 修改

阅读量230

点赞数

文章标签：算法 leetcode

于 2023-09-21 03:53:41 首次发布

本文链接：https://blog.csdn.net/u013003382/article/details/133115414

版权

文章讲述了如何使用单调栈解决两个编程问题：寻找每日温度序列中每个元素右边第一个更大的元素和查找数组中每个元素右边第一个比其大的元素。通过分析三种情况，利用单调栈的特性优化了时间复杂度。

摘要由CSDN通过智能技术生成

739. 每日温度

代码随想录

初步思路：暴力解法，两层循环

总结：

单调栈通常是一维数组，要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置

使用单调栈主要有三个判断条件。

当前遍历的元素T[i]小于栈顶元素T[st.top()]的情况: 此时满足递增栈（栈头到栈底的顺序），所以直接入栈.
当前遍历的元素T[i]等于栈顶元素T[st.top()]的情况: 依然直接入栈，因为我们要求的是右边第一个比自己大的元素，而不是大于等于.
当前遍历的元素T[i]大于栈顶元素T[st.top()]的情况:

        answer = [0]*len(temperatures)
        stack = [0]
        for i in range(1,len(temperatures)):
            # 情况一和情况二
            if temperatures[i]<=temperatures[stack[-1]]:
                stack.append(i)
            # 情况三
            else:
                while len(stack) != 0 and temperatures[i]>temperatures[stack[-1]]:
                    answer[stack[-1]]=i-stack[-1]
                    stack.pop()
                stack. Append(i)

用时：45分钟

496.下一个更大元素 I

代码随想录

初步思路：单调栈

总结：

情况一：当前遍历的元素T[i]小于栈顶元素T[st.top()]的情况

此时满足递增栈（栈头到栈底的顺序），所以直接入栈。

情况二：当前遍历的元素T[i]等于栈顶元素T[st.top()]的情况

如果相等的话，依然直接入栈，因为我们要求的是右边第一个比自己大的元素，而不是大于等于！

情况三：当前遍历的元素T[i]大于栈顶元素T[st.top()]的情况

此时如果入栈就不满足递增栈了，这也是找到右边第一个比自己大的元素的时候。

判断栈顶元素是否在nums1里出现过，（注意栈里的元素是nums2的元素），如果出现过，开始记录结果。

记录结果这块逻辑有一点小绕，要清楚，此时栈顶元素在nums2数组中右面第一个大的元素是nums2[i]（即当前遍历元素）。

        result = [-1]*len(nums1)
        stack = [0]
        for i in range(1,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:
                        index = nums1.index(nums2[stack[-1]])
                        result[index]=nums2[i]
                    stack.pop()                 
                stack. Append(i)

用时：45分钟