牛客OJ：可以返回当前最小值的栈

最新推荐文章于 2020-05-12 22:23:56 发布

ShellDawn

最新推荐文章于 2020-05-12 22:23:56 发布

阅读量158

点赞数

分类专栏： NewCoder OJ 文章标签：牛客OJ 最小栈

本文链接：https://blog.csdn.net/ShellDawn/article/details/88910047

版权

NewCoder OJ 专栏收录该内容

37 篇文章 0 订阅

订阅专栏

维护两个栈：

#include <bits/stdc++.h>
using namespace std;

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

int main(){
    Solution S;
    S.push(3);
    printf("%d\n",S.min());
    S.push(4);
    printf("%d\n",S.min());
    S.push(2);
    printf("%d\n",S.min());
    S.push(1);
    printf("%d\n",S.min());
    S.pop();
    printf("%d\n",S.min());
    S.pop();
    printf("%d\n",S.min());
    S.push(0);
    printf("%d\n",S.min());
    return 0;
}