O(1)时间复杂度求栈的最小元素

题目:
   定义栈的数据结构,要求添加一个min函数,能够得到栈的最小元素。要求函数min、push以及pop的时间复杂度都是O(1)。
 
思路:
   栈数据结构的特点是先进后出,push和pop操作都是对栈顶元素进行的操作,所以都是常数时间复杂度,即O(1),但是由于栈中数据是无序的,所以仅仅通过栈是无法得到最小元素的,又由于每次出栈和进栈都可能会使栈的最小元素发生变化,所以考虑用一个辅助的栈空间来存储每次进栈和出栈之后的最小元素,为了节省空间,辅助栈中不存储最小元素的值,而存储最小元素的地址或者在数据栈中的索引。
 
代码:
#include <vector>
using namespace std;
template<typename T> class CStack
{
public:
 CStack();
 ~CStack();
 
 void pop(T& value);
 void push(const T& value);
 const T& getMin() const;
private:
 vector<T> _data;
 vector<size_t> _minIndex;
}
template<typename T> CStack<T>::CStack()
{
}
template<typename T> CStack<T>::~CStack()
{
}
template<typename T> void CStack<T>::pop(T& value)
{
 if(_data.empty())
 {
  return;
 }
 else
 {
  value = _data.back();
  _data.pop_back();
  _minIndex.pop_back();
 }
}


template<typename T> void CStack<T>::push(T& value)
{
 _data.push_back(value);
 
 if(_minIndex.empty())
 {
  _minIndex.push_back(0);
 }
 else if(value < _data[_minIndex.back()])
 {
  _minIndex.push_back(_data.size()-1);
 }
 else
 {
  _minIndex.push_back(_minIndex.back());
 }
}
template<typename T> const T&CStack<T>::getMin() const
{
 assert( !_data.empty() );
 assert( !_minIndex.empty() );
 
 return _data[_minIndex.back()];
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值