剑指Offer第二十题:包含min函数的栈

版权声明:欢迎交流,本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/weixin_42513339/article/details/89052268

题目描述

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

思路:唉,一开始写了个链表虽然能成功,但是还是太麻烦,这里应该直接用栈来做,STL是神器,直接利用两个stack,一个存数,一个存最小数。

存最小数的栈原理是,现存入一个数,第二数如果比top顶上的小,那么继续存入,存数字的栈弹出的时候如果弹出的元素的值跟存最小数的top值相等,那么存最小数的栈也弹出。

代码如下

class Solution {
public:
    void push(int value) {
        va.push(value);
        if(myMin.empty())
            myMin.push(value);
        else if(value < myMin.top())
            myMin.push(value);
    }
    void pop() {
        if(!va.empty())
        {
            if(va.top() == myMin.top())
                myMin.pop();
            va.pop();  
        }               
    }
    int top() {
        return va.top();
    }
    int min() {
        return myMin.top();
    }
    
private:
    stack<int>va;
    stack<int>myMin;
};

猜你喜欢

转载自blog.csdn.net/weixin_42513339/article/details/89052268
今日推荐