程序员面试金典(9)设计一个栈,返回最小值min

来源:互联网 发布:bootcamp for windows 编辑:程序博客网 时间:2024/06/17 01:25
#include <iostream>
#include <string>
#include <map>
#include <stack>
using namespace std;

template <typename T> class StackWithMin
{
public:
    StackWithMin(void) {}
    T& top(void);
    void push(const T& value);
    void pop(void);
    const T& min(void) const;
    bool empty() const;
    size_t size() const;
private:
    std::stack<T>   m_data;    
    std::stack<T>   m_min;     
};

template <typename T> void StackWithMin<T>::push(const T& value)
{
    m_data.push(value);
    if(m_min.size() == 0 || value < m_min.top())
        m_min.push(value);
    else
        m_min.push(m_min.top());
}

template <typename T> void StackWithMin<T>::pop()
{
    assert(m_data.size() > 0 && m_min.size() > 0);
    m_data.pop();
    m_min.pop();
}

template <typename T> const T& StackWithMin<T>::min() const
{
    assert(m_data.size() > 0 && m_min.size() > 0);
    return m_min.top();
}

template <typename T> T& StackWithMin<T>::top()
{
    return m_data.top();
}

template <typename T> bool StackWithMin<T>::empty() const
{
    return m_data.empty();
}

template <typename T> size_t StackWithMin<T>::size() const
{
    return m_data.size();
}

int main()
{
    
    //getchar();
    return 0;
}

0 0
原创粉丝点击