Leetcode503
1.问题描述
2.解决方案
1.这道题和【Leetcode单调栈】739. 每日温度(有关于单调栈类型题的深入理解概括!!)也几乎如出一辙。不同的时候本题要循环数组了。
2.相信不少同学看到这道题,就想那我直接把两个数组拼接在一起,然后使用单调栈求下一个最大值不就行了!确实可以!将两个nums数组拼接在一起,使用单调栈计算出每一个元素的下一个最大值,最后再把结果集即result数组resize到原数组大小就可以了。
3.其实也可以不扩充nums,而是在遍历的过程中模拟走了两边nums。
解法一:繁琐写法
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
//1.准备递减栈
stack<int> stack;
vector<int> ans(nums.size(),-1);
if(nums.size()<2) return ans;
//2.遍历构建单调栈
stack.push(0);
for(int j=1;j<=2*nums.size()-1;j++){
//遍历两遍数组
int i=j%nums.size();
//情况1,2
if(nums[stack.top()]>=nums[i]){
stack.push(i);
continue;
}
//情况3:出结果的一种情况
if(nums[stack.top()]<nums[i]){
while(stack.empty()== false&&nums[stack.top()]<nums[i]){
ans[stack.top()]=nums[i];
stack.pop();
}
stack.push(i);
continue;
}
}
return ans;
}
};
解法二:单调栈简洁写法(以后都这么写!!)
class Solution1{
public:
vector<int> nextGreaterElements(vector<int>& nums) {
//1.准备阶段
stack<int> st;
vector<int> result(nums.size(), -1);
if(nums.size()==0) return result;
//2.遍历构造单调栈阶段
for(int i=0;i<nums.size()*2;i++){
//模拟遍历两边nums,注意一下都是用i % nums.size()来操作
//情况1,2,3->合一起写了很简洁!
while(st.empty()== false&&nums[st.top()]<nums[i%nums.size()]){
result[st.top()]=nums[i % nums.size()];
st.pop();
}
st.push(i % nums.size());
}
return result;
}
};