Day 52 卡玛笔记

这是基于代码随想录的每日打卡

739. 每日温度

给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替。

示例 1:

复制代码
输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]

示例 2:

复制代码
输入: temperatures = [30,40,50,60]
输出: [1,1,1,0]

示例 3:

复制代码
输入: temperatures = [30,60,90]
输出: [1,1,0]

单调栈

寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置,此时我们就要想到可以用单调栈了

就是用一个栈来记录我们遍历过的元素,单调栈里只需要存放遍历过的下标就可以了。

先将第一个元素放进栈中,然后遍历后面的元素,如果遍历到的当前元素比栈口元素小,则继续放入栈中;

如果比栈口元素大,则说明栈口元素找到了第一个比它的值大的元素,符合题意,则弹出栈口元素,同时将两个元素下标对应相减,就是栈口元素的结果值。

还没完,如果栈口元素弹出后,下一个栈口元素还比当前元素小,则继续弹出操作且下标相减存入结果。

这样就保证单调栈里面的值是单调递减,且可以确定栈口是朝右,这样比较容易想象。

栈里保存的是我们的下标值

代码

python 复制代码
class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        # 单调栈
        stack=[0]
        # 存放结果
        res=[0]*len(temperatures)
        # 遍历温度
        for i in range(1,len(temperatures)):
            # 当前元素比栈口元素小,则存入
            if temperatures[i]<=temperatures[stack[-1]]:
                stack.append(i)
            # 当前元素比栈口元素大,则一直弹栈直到不符合条件
            else:
                while stack and temperatures[i]>temperatures[stack[-1]]:
                    index=stack.pop()
                    res[index]=i-index
                stack.append(i)
        return res

运行结果


496. 下一个更大元素 I

nums1 中数字 x下一个更大元素 是指 xnums2 中对应位置 右侧第一个x 大的元素。

给你两个 没有重复元素 的数组 nums1nums2 ,下标从 0 开始计数,其中nums1nums2 的子集。

对于每个 0 <= i < nums1.length ,找出满足 nums1[i] == nums2[j] 的下标 j ,并且在 nums2 确定 nums2[j]下一个更大元素 。如果不存在下一个更大元素,那么本次查询的答案是 -1

返回一个长度为 nums1.length 的数组 ans 作为答案,满足 ans[i] 是如上所述的 下一个更大元素

示例 1:

复制代码
输入:nums1 = [4,1,2], nums2 = [1,3,4,2].
输出:[-1,3,-1]
解释:nums1 中每个值的下一个更大元素如下所述:
- 4 ,用加粗斜体标识,nums2 = [1,3,4,2]。不存在下一个更大元素,所以答案是 -1 。
- 1 ,用加粗斜体标识,nums2 = [1,3,4,2]。下一个更大元素是 3 。
- 2 ,用加粗斜体标识,nums2 = [1,3,4,2]。不存在下一个更大元素,所以答案是 -1 。

示例 2:

复制代码
输入:nums1 = [2,4], nums2 = [1,2,3,4].
输出:[3,-1]
解释:nums1 中每个值的下一个更大元素如下所述:
- 2 ,用加粗斜体标识,nums2 = [1,2,3,4]。下一个更大元素是 3 。
- 4 ,用加粗斜体标识,nums2 = [1,2,3,4]。不存在下一个更大元素,所以答案是 -1 。

单调栈

跟上题一样,构造一个单调栈,栈口朝右,且保证栈内元素单调递减

先将nums2的第一个元素存入栈中,随后遍历nums2的后续元素

遍历过程发现比栈口元素小,则加进栈。

遍历过程发现有比栈口元素大的元素,则将栈口元素弹出,且检查栈口元素有没有在nums1出现过,若有则将res数组对应下标的值修改为当前遍历到的元素,且持续检查栈口元素与当前元素的大小。

代码

python 复制代码
class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        # 存放结果
        res=[-1]*len(nums1)
        # nums2的单调栈,存放的是nums2元素的下标
        stack=[0]
        for i in range(1,len(nums2)):
            # 若当前元素比栈口元素小则进栈,保证单调递减
            if nums2[i]<=nums2[stack[-1]]:
                stack.append(i)
            else:
                # 若当前元素比栈口元素大则弹出栈口元素
                while stack and nums2[i]>nums2[stack[-1]]:
                    # 栈口元素的下标
                    index=stack.pop()
                    # 如果栈口元素在nums2出现过
                    if nums2[index] in nums1:
                        # nums2[index]是栈口元素
                        # nums1.index[nums2(index)]是栈口元素在nums1出现的位置
                        res[nums1.index(nums2[index])]=nums2[i]
                stack.append(i)
        return res

运行结果


503. 下一个更大元素 II

给定一个循环数组 numsnums[nums.length - 1] 的下一个元素是 nums[0] ),返回 nums 中每个元素的 下一个更大元素

数字 x下一个更大的元素 是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1

示例 1:

复制代码
输入: nums = [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数; 
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。

示例 2:

复制代码
输入: nums = [1,2,3,4,3]
输出: [2,3,4,-1,4]

单调栈

只需要把两个nums拼接,再按单调栈的规矩遍历两个nums

不过不用开辟一个二倍长度的数组,只需要对下标取模

注意栈口元素和取结果数组下标时都要取模

代码

python 复制代码
class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        # 单调栈(栈口朝右,单调递减),存储的依旧是下标
        stack=[0]
        # 数组长度
        lens=len(nums)
        # 存放结果
        res=[-1]*lens
        for i in range(1,lens*2):
            # 遍历到的当前元素小于栈口元素
            if nums[i%lens]<=nums[stack[-1]%lens]:
                stack.append(i)
            # 遍历到的元素大于栈口元素,则一直弹栈,直到栈口元素小于当前元素
            else:
                while stack and nums[stack[-1]%lens]<nums[i%lens]:
                    index=stack.pop()
                    res[index%lens]=nums[i%lens]
                stack.append(i%lens)
        return res

运行结果

相关推荐
维度攻城狮2 小时前
实现在Unity3D中仿真汽车,而且还能使用ros2控制
python·unity·docker·汽车·ros2·rviz2
简简单单做算法2 小时前
基于mediapipe深度学习和限定半径最近邻分类树算法的人体摔倒检测系统python源码
人工智能·python·深度学习·算法·分类·mediapipe·限定半径最近邻分类树
hvinsion3 小时前
基于PyQt5的自动化任务管理软件:高效、智能的任务调度与执行管理
开发语言·python·自动化·自动化任务管理
Tisfy3 小时前
LeetCode 2360.图中的最长环:一步一打卡(不撞南墙不回头) - 通过故事讲道理
算法·leetcode··题解
LuckyAnJo3 小时前
Leetcode-100 链表常见操作
算法·leetcode·链表
飞飞翼4 小时前
python-flask
后端·python·flask
双叶8364 小时前
(C语言)虚数运算(结构体教程)(指针解法)(C语言教程)
c语言·开发语言·数据结构·c++·算法·microsoft
工一木子4 小时前
大厂算法面试 7 天冲刺:第5天- 递归与动态规划深度解析 - 高频面试算法 & Java 实战
算法·面试·动态规划
林九生5 小时前
【Python】Browser-Use:让 AI 替你掌控浏览器,开启智能自动化新时代!
人工智能·python·自动化
猿界零零七6 小时前
执行paddle.to_tensor得到全为0
python·paddle