带有O(1)时间返回最小值的stack

最新推荐文章于 2021-05-13 14:54:31 发布

逍遥游07

最新推荐文章于 2021-05-13 14:54:31 发布

阅读量588

点赞数

分类专栏：算法导论

本文链接：https://blog.csdn.net/kesonyk/article/details/47002559

版权

算法导论专栏收录该内容

30 篇文章 0 订阅

订阅专栏

本文介绍了一种特殊的数据结构——栈，该栈在进行常规压栈和弹栈操作的同时，能够高效地获取栈内元素的最小值。通过维护两个栈：一个用于存储所有元素（data），另一个用于跟踪最小值（minVal），实现了在O(1)时间内查询最小元素的功能。

摘要由CSDN通过智能技术生成

主要是在辅助的栈中push当前最小的值，若data栈pop出的数不是当前最小的，则辅助栈不动；若data栈pop出当前最小的值，则辅助栈也pop

#include <vector>
#include <iostream>

using namespace std;

template<typename object>
class minStack
{
public:
	void push(object val)
	{
		data.push_back(val);
		if(minVal.empty()||val<＝minVal.back()) //需要等号
			minVal.push_back(val);
	}

	void pop()
	{
		if(data.back()==minVal.back())
			minVal.pop_back();
		data.pop_back();
	}

	void printMin()
	{
		if(minVal.empty())
			cout<<"None element"<<endl;
		else
			cout<<minVal.back()<<endl;

	}
private:
	vector<object> data;
	vector<object> minVal;
};


int main()
{
	minStack<int> mystack;
	mystack.push(5);
	mystack.printMin();  //5
	mystack.push(4);
	mystack.printMin();  //4
	mystack.push(6);
	mystack.printMin();  //4
	mystack.push(3);
	mystack.printMin();  //3
	mystack.pop();
	mystack.printMin();  //4
}