题解 | #包含min函数的栈#

包含min函数的栈

http://www.nowcoder.com/practice/4c776177d2c04c2494f2555c9fcc1e49

解题思路

使用一个辅助栈存储最小值,min函数直接返回辅助栈顶就可以。


代码实现

class Solution {
public:
    stack<int> normal, minval;
    void push(int value) {
        normal.push(value);
        if (minval.empty()) {
            minval.push(value);
        }
        else {
            if (value <= minval.top()) {
                minval.push(value);
            }
            else {
                minval.push(minval.top());
            }
        }
    }
    void pop() {
        if(normal.empty()) return;
        normal.pop();
        minval.pop();
    }
    int top() {
        if(normal.empty()) return -1;
        return normal.top();
    }
    int min() {
        return minval.top();
    }
};

注意

不能想着辅助minval栈只入最小值,比如minval栈顶是1,来个元素6就不入栈了,这样的话在normal栈出栈的时候两者就不能保持同步了,结果自然会出错。

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务