实现O(1)时间复杂度带有min和max 函数的栈

最新推荐文章于 2022-02-01 21:33:14 发布

瑞卡哥哥

最新推荐文章于 2022-02-01 21:33:14 发布

阅读量3.6k

点赞数

分类专栏：数据结构 [源码分享] 算法 C++

本文链接：https://blog.csdn.net/zhouqinghe24/article/details/40558195

版权

只是演示实现，不考虑栈使用的数据结构是vector 还是其他容器。

代码如下

#include <iostream>
#include <vector>

using namespace std;

template <class T>
class minMaxStack
{
public:
	minMaxStack()
	{
		DataStack = new vector<T>;
		minStack = new vector<int>;
		maxStack = new vector<int>;
	}
	~minMaxStack()
	{
		delete DataStack;
		delete minStack;
		delete maxStack;
	}
	bool isEmpty()
	{
		return DataStack->size() == 0;
	}
	void push(T data)
	{
		
		if (isEmpty())
		{
			DataStack->push_back(data);
			minStack->push_back(0);
			maxStack->push_back(0);
		}
		else
		{
			DataStack->push_back(data);
			int minIndex = *(minStack->end()-1);
			int maxIndex = *(maxStack->end()-1);
			int min = *(DataStack->begin() + minIndex);
			int max = *(DataStack->begin() + maxIndex);
			if (min > data)
			{
				minStack->push_back(DataStack->size() - 1)