1.每日温度
代码:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
stack<int> st;
vector<int> result(temperatures.size(), 0);
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;
}
};
思路:
单调栈通常用来解决“找出后续元素中以一个比当前元素大的值(递增单调栈)”或“第一个比当前元素小的值(单调递减栈)”——这里的单调顺序指的是从栈顶到栈底元素的大小关系。
具体的使用原理如下(以此题为例):
有点像消消乐,或者对对碰——遍历数组,如果当前元素>栈顶元素,那就把比它小的元素全部“碰”走。在这个“碰”的过程里,灰溜溜地出栈的元素其实就是在这个时候第一次遇到了比它大的元素。这就给出了我们上述提出的问题的回答。
(感觉也像是打擂台赛一样,挑战完比你水平低的人,让它从这个排行榜上下来,直到你也遇到了比你水平高的人,就说明你的水平就在这了。“没想到,我还是第一次遇到如此优秀的后辈!”)
要注意!!在这个“碰”的过程,也就是代码里当前元素大于栈顶元素的时候,是一个循环的过程!!!
还要此题也是比较取巧,直接把下标压入栈,在比较的时候通过索引找元素值来比。
2.下一个更大元素1
代码:
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
stack<int> st;
vector<int> result(nums1.size(), -1);
unordered_map<int,int> umap;
if(nums1.size() == 0) return result;
for(int i = 0; i < nums1.size(); i++){
umap[nums1[i]] = i;
}
st.push(nums2[0]);
for(int i = 0; i < nums2.size(); i++){
if(nums2[i] <= st.top()){
st.push(nums2[i]);
}else{
while(!st.empty() && nums2[i] > st.top()){
if(umap.count(st.top()) > 0){
result[umap[st.top()]] = nums2[i];
}
st.pop();
}
st.push(nums2[i]);
}
}
return result;
}
}
note:上一题的升级版,其实还是那个套路,只是收集结果有所差异。我们在遇到弹出的元素是nums1里的元素时才收集,而且我们还得直到这个数对应的下标——那就得去进行判断,还得去通过元素找下标——不就是用map来记录吗?下标也可以用key-value的形式找到。
3.下一个更大元素2
代码:
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
stack<int> st;
int size = nums.size();
vector<int> result(nums.size(),-1);
if(nums.size() == 0) return result;
st.push(0);
for(int i = 1; i < 2 * nums.size(); i++){
if(nums[i % size] <= nums[st.top()]){
st.push(i % size);
}else{
while(!st.empty() && nums[i % size] > nums[st.top()]){
result[st.top()] = nums[i % size];
st.pop();
}
st.push(i % size);
}
}
return result;
}
};
note:
套路一样,就是涉及到环,可以考虑用取余的方式来进行遍历数组。