数据结构算法 | 单调栈
2021/10/28 22:09:54
本文主要是介绍数据结构算法 | 单调栈,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
文章目录
- 算法概述
- 题目
- 下一个更大的元素 I
- 思路
- 代码
算法概述
当题目出现 「找到最近一个比其大的元素」 的字眼时,自然会想到 「单调栈」 。——三叶姐
单调栈以严格递增or递减的规则将无序的数列进行选择性排序。
题目
下一个更大的元素 I
给你两个 没有重复元素 的数组 nums1
和 nums2
,其中 nums1
是 nums2
的子集。(题源力扣)
请你找出 nums1
中每个元素在 nums2
中的下一个比其大的值。
nums1
中数字 x
的下一个更大元素是指 x
在 nums2
中对应位置的右边的第一个比 x
大的元素。如果不存在,对应位置输出 -1
。
示例:
输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
输出: [-1,3,-1]
解释:
- 对于 num1 中的数字 4 ,你无法在第二个数组中找到下一个更大的数字,因此输出 -1 。
- 对于 num1 中的数字 1 ,第二个数组中数字 1 右边的下一个较大数字是 3 。
- 对于 num1 中的数字 2 ,第二个数组中没有下一个更大的数字,因此输出 -1 。
思路
- 倒序遍历
nums2
用stack
暂存严格递增的数据,即,若 栈不为空 且 当前遍历到的元素大于栈顶元素 则弹出栈顶元素。 - 用
unordred_map
存储 遍历到的元素(key
) 和 最近一个比其大的元素(value
) 之间的关系。经过步骤一后,若栈不为空,则栈顶元素即为value
,否则,value
为-1
。 - 正序遍历
nums1
并依赖unordred_map
得到题目要求的结果数组。
代码
class Solution { public: vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) { unordered_map<int, int> m; stack<int> st; for(int i = nums2.size()-1; i >= 0; i--){ int t = nums2[i]; while(st.size() && t>st.top()) st.pop(); m[t] = st.empty() ? -1 : st.top(); st.push(t); } vector<int> v; for(int i = 0; i < nums1.size(); i++){ v.push_back(m[nums1[i]]); } return v; } };
这篇关于数据结构算法 | 单调栈的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-10-01基于Python+Vue开发的医院门诊预约挂号系统
- 2024-10-01基于Python+Vue开发的旅游景区管理系统
- 2024-10-01RestfulAPI入门指南:打造简单易懂的API接口
- 2024-10-01初学者指南:了解和使用Server Action
- 2024-10-01Server Component入门指南:搭建与配置详解
- 2024-10-01React 中使用 useRequest 实现数据请求
- 2024-10-01使用 golang 将ETH账户的资产平均分散到其他账户
- 2024-10-01JWT用户校验课程:从入门到实践
- 2024-10-01Server Component课程入门指南
- 2024-09-30Dnd-Kit学习:新手快速入门指南