剑指offer——包含min函数的栈

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数。

 

#include<cstdlib>
#include<stack>
#include<assert.h>
using namespace std;
class Solution {
public:
    stack<int> m_data, m_min;
    void push(int 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());
    }
    void pop() {
        assert(m_data.size() > 0 && m_min.size() > 0);
        m_data.pop();
        m_min.pop();
    }
    int top() {
        return m_data.top();
    }
    int min() {
        assert(m_data.size() > 0 && m_min.size() > 0);
        return m_min.top();
    }
};

得到的