码迷,mamicode.com
首页 > 其他好文 > 详细

O(1)时间求栈中最小(大)元素

时间:2014-09-06 11:02:43      阅读:172      评论:0      收藏:0      [点我收藏+]

标签:   笔试面试   最大值   最小值   

问题:对stack进行扩展,完成正常的push,pop操作,新增访问最小(大)元素的接口min(max),使得push,pop,Min的时间复杂度都是O(1)。

难点在于怎么维持stack的最小(大)值,一切排序和查找都不可能实现O(1)的时间复杂度找到最小值。

思路:如下图所示,以空间换取时间。通过增加一个最小值栈来存储上一个最小值,以维持目前的最小值。

bubuko.com,布布扣

1、  入栈的元素比当前的min小:如当min=3时,元素2入栈,则将当前最小值3push进最小值栈,min变为2。

2、  出栈的元素为当前的min:如当min=1时,元素1出栈,则将元素1出栈,最小值栈的元素2出栈,min=2。即回到了上一个状态。

以上是在时间O(1)下求最小值,最大值同样如此,利用额外的栈就可实现。


代码实现:

#include<vector>
#include<iostream>
#include<assert.h>
usingnamespace std;
 
template<typenameT>
classCStack{
public:
         CStack():min_elem(0){}
         ~CStack(){}
        T&pop();
         void push(const T& value);
        Tget_min() const;
private:
         vector<T> _data;
         vector<size_t> _minstore;
         T min_elem;
};
 
template<typenameT>
T&CStack<T>::pop(){
         assert( !_data.empty() );
         T value;
                   value = _data.back();
                   _data.pop_back();
                  if(value== min_elem){
                            min_elem =_minstore.back();
                            _minstore.pop_back();
                   }
         return value;
}
 
template<typenameT>
voidCStack<T>::push(const T& value){
         if(_data.empty()){
                   min_elem = value;
         }
         else if(value <= min_elem){
                   _minstore.push_back(min_elem);
                   min_elem = value;
         }
         _data.push_back(value);
}
template<typenameT>
TCStack<T>::get_min() const{
         assert( !_data.empty() );
         return min_elem;
}
intmain(){
         CStack<int> s;
         s.push(3);
         int min = s.get_min();
         cout<<"current min_elem:"<<min<<endl;
         s.push(4);
         s.push(5);
         min = s.get_min();
         cout<<"current min_elem:"<<min<<endl;
 
         s.push(2);
         min = s.get_min();
         cout<<"current min_elem:"<<s.get_min()<<endl;
    s.push(2);
         min = s.get_min();
         cout<<"current min_elem:"<<min<<endl;
         s.pop();
         cout<<"current min_elem:"<<s.get_min()<<endl;
 
    s.push(1);
         cout<<"current min_elem:"<<s.get_min()<<endl;
 
         return 0;
}


O(1)时间求栈中最小(大)元素

标签:   笔试面试   最大值   最小值   

原文地址:http://blog.csdn.net/ywok526/article/details/39099097

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!