题目:739. 每日温度
文章链接:代码随想录
视频链接:LeetCode:739.每日温度
题目链接:力扣题目链接
图释:通常是一堆数组,要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置,此时考虑单调栈
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
// 初始化结果数组
vector<int> result(temperatures.size(), 0);
// 定义单调栈
stack<int>st;
st.push(0); //把首个元素的下标添加到栈里
for(int i=1; i<temperatures.size(); i++){
if(temperatures[i] <= temperatures[st.top()]){ // 如果栈顶元素小于遍历元素
st.push(i); // 放入栈中
}
else{
while(!st.empty()&&temperatures[i] > temperatures[st.top()]){ // 栈内不能为空,并且循环拿比遍历数大
result[st.top()] = i-st.top(); //添加到结果集中,即栈中弹出的数与当前的下标进行比较
st.pop(); //将记录后的结果进行弹出
}
st.push(i); //将当前的元素放进栈中
}
}
return result;
}
};
题目:496.下一个更大元素 I
文章链接:代码随想录
题目链接:力扣题目链接
图释:
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
// 定义单调栈
stack<int>st;
// 初始化结果集
vector<int> result(nums1.size(), -1); // 以nums1的元素存放结果
// 定义map映射
unordered_map<int, int>umap; // key:元素值 value:下标
for(int i=0; i<nums1.size(); i++){
umap[nums1[i]]=i; // 遍历nums1中的每个元素,建立对应关系
}
// 将首元素放入栈中
st.push(0);
for(int i=1; i<nums2.size(); i++){ // 遍历nums2数组
if(nums2[i] <= nums2[st.top()]){ // 栈顶元素比党鞭
st.push(i); //持续放入
}
else{
while(!st.empty() && nums2[i]>nums2[st.top()]){ //栈内元素不为空,且当前元素比栈顶元素大
if(umap.count(nums2[st.top()])>0){ // 查看map中是否存在该元素 .count()与.find()函数查找键是都在map中
int index = umap[nums2[st.top()]]; //根据map找到nums2[st.top()] 在 nums1中的下标
result[index] = nums2[i]; // 赋值下一个更大元素
}
st.pop();
}
st.push(i);
}
}
return result;
}
};