牛客网 《剑指Offer》编程 20.包含min函数的栈

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。

基本思路

第一种思路：在数值入栈的时候记录最小值。最后返回最小值即可。但是如果最小值弹出了，不能返回第二小的值，所以没什么用。

第二种思路：元素正常入主栈。

使用一个辅助栈，在入栈时，如果辅助栈为空或者当前元素的值小于辅助栈栈顶值，则将当前元素压入辅助栈；

如果辅助站不为空，当前元素值大于辅助栈栈顶值，则将辅助栈栈顶的值再次压栈。

代码实现

class Solution {
public:
    stack<int> stk;
    stack<int> minstk;
    
    void push(int value) {
        //压栈操作：stk直接压栈
        //minstk压栈方式：如果当前树小于minstk栈顶数，则直接压栈；
        //如果当前数大于minstk栈顶数，则将minstk栈顶数压栈。
        stk.push(value);
        if(minstk.empty()||value<minstk.top()){
            minstk.push(value);
        }else{
            minstk.push(minstk.top());
        }
    }
    void pop() {
        //stk与minstk是同时压栈和同时弹栈，因此只要判断一个不为空，则同时不为空，可以弹栈。
        if(!stk.empty()){
            stk.pop();
            minstk.pop();
        }
    }
    int top() {
        //返回stk当前栈顶值。
        return stk.top();
    }
    int min() {
        //返回最小元素值
        return minstk.top();
    }
};