2.设计包含min函数的栈。

最新推荐文章于 2023-07-24 10:28:56 发布

转载最新推荐文章于 2023-07-24 10:28:56 发布 · 595 阅读

数据结构与算法专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种特殊栈的设计方案，该栈除了具备常规的push和pop操作外，还提供了一个min函数，用于返回栈中当前的最小值。所有操作（min、push及pop）的时间复杂度均为O(1)。文章通过使用两个辅助栈，一个存储所有数据元素，另一个跟踪最小值的变化，从而实现了这一功能。

2.设计包含min函数的栈。
定义栈的数据结构，要求添加一个min函数，能够得到栈的最小元素。
要求函数min、push以及pop的时间复杂度都是O(1)。

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

template <typename T>
class StackSuppliedMin{
private:
	vector<T> dates;
	vector<size_t> minSta;
public:
	void push(T date);
	void pop();
	T min();
	void display();
};

template <typename T> void StackSuppliedMin<T>::push(T date){
	dates.push_back(date);
	if(minSta.empty() || date <= dates[minSta.back()])
		minSta.push_back(dates.size()-1);
}
template <typename T> void StackSuppliedMin<T>::pop(){
	assert(!dates.empty());
	if(dates.back() == dates[minSta.back()])
		minSta.pop_back();
	dates.pop_back();
}
template <typename T> T StackSuppliedMin<T>::min(){
	assert(!dates.empty() && !minSta.empty());
	return dates[minSta.back()];
}
template <typename T> void StackSuppliedMin<T>::display(){
	cout << min() << endl;
}

int main(){
	StackSuppliedMin<int> iStack;
	iStack.push (6);
	iStack.push (7);
	iStack.push (4);
	iStack.push (5);
	iStack.push (4);
	iStack.push (2);
	iStack.push (8);
	iStack.display();
	iStack.pop();
	iStack.display();
	iStack.pop();
	iStack.display();
	iStack.pop();
	iStack.display();
	iStack.pop();
	iStack.display();
	iStack.pop();
	iStack.display();
	iStack.pop();
	iStack.display();
	iStack.pop();
	return 0;
}