20、剑指offer--包含min函数的栈

题目描述

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

 

解题思路：

1）本题定义一个数据栈m_data，正常的push、pop、top操作

2）定义辅助栈m_min保证每次栈顶元素都是最小元素，且和数据栈中元素的数目相同

#include <iostream>
#include <stack>
using namespace std;
class Solution {
public:
    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() {
        m_data.pop();
        m_min.pop();
    }
    int top() {
        return m_data.top();
    }
    int min() {
        return m_min.top();
    }
private:
    stack<int> m_data;//数据栈
    stack<int> m_min;//辅助栈
};

int main()
{
    Solution s;
    s.push(3);
    s.push(4);
    s.push(5);
    s.push(1);
    s.push(2);
    cout<<"最小值为:"<<s.min()<<endl;
    s.pop();
    s.pop();
    cout<<"最小值为"<<s.min()<<endl;
}