经典算法——包含min函数的栈

Posted 2020-06-20 zhihua_bupt

题目描述

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

class Solution {
public:
    stack<int> stack1;
    stack<int> helpStack;//辅助栈，存放最小值
    
    void push(int value) 
    {
        stack1.push(value);
        
        if(helpStack.size()==0 || value<helpStack.top())//每次都把最小值压入辅助栈
            helpStack.push(value);
        else
            helpStack.push(helpStack.top());
    }        

    void pop() 
    {
         stack1.pop();
         helpStack.pop();
    }
    
    int top() 
    {
        return stack1.top();
    }
    
    int min() 
    {
        return helpStack.top();
    }
};