0
点赞
收藏
分享

微信扫一扫

下一个更大的元素(单调栈)

左小米z 2022-02-12 阅读 72

文章目录

单调栈

单调栈的介绍及应用

单调栈的单调增和单调减都是从栈顶至栈底来看的。

#include <stack>
#include <vector>
#include <iostream>
using namespace std;

stack<int, vector<int>> get_monotoneStack(vector<int>& arr);

int main() {
	vector<int> arr{10,2,9,4,7,3,5 };
	stack<int, vector<int>> monotoneStack = get_monotoneStack(arr);
	while (!monotoneStack.empty()) {
		cout << monotoneStack.top() << endl;
		monotoneStack.pop();
	}
	return 0;
}

// 递增
stack<int,vector<int>> get_monotoneStack(vector<int>& arr) {
	stack<int,vector<int>> monotoneStack;
	for (auto& a : arr) {
		// 重点是这一个循环,每个元素是必须要push进去的。
		// 如果栈里面的元素没有比它大的了,那么栈清空,它进去。
		// 如果栈里面的元素有比它大的,那直接进去,该元素是离它最近的比它大的元素。
		while (!monotoneStack.empty() && a >= monotoneStack.top())
			monotoneStack.pop();
		monotoneStack.push(a);
	}
	return monotoneStack;
}

题目

下一个更大的元素

vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        map<int,int> location={};
        stack<int,vector<int>> monotoneStack={};
        for(int i = nums2.size() - 1;i >= 0;i--){
            while(!monotoneStack.empty() && nums2[i] >= monotoneStack.top()){
                monotoneStack.pop();
            }
            if (monotoneStack.empty()) location[nums2[i]] = -1;
            else location[nums2[i]] = monotoneStack.top();
            monotoneStack.push(nums2[i]);   
        }
        vector<int> res={};
        for(auto &n:nums1)
            res.push_back(location[n]);
        return res;
    }
举报

相关推荐

0 条评论