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

双栈维护

最新推荐文章于 2025-07-20 15:11:01 发布

原创最新推荐文章于 2025-07-20 15:11:01 发布 · 187 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#牛客OJ #最小栈

NewCoder OJ 专栏收录该内容

37 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现带有最小值查询功能的数据结构的方法。通过维护一个辅助栈，可以在常数时间内获取到当前栈中的最小元素。这种方法适用于需要频繁查询最小值的场景。

维护两个栈：

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