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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ShellDawn/article/details/88910047

维护两个栈:

#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;
}


猜你喜欢

转载自blog.csdn.net/ShellDawn/article/details/88910047