牛客题解 | 最小栈

题目

题目链接

题解：

考察点： 栈

易错点：

本题要求\(min\)、\(push\)、\(pop\)三种操作都在\(O(1)\)复杂度内完成，也就是说都必须在常数时间内完成，其中\(push\)和\(pop\)都是栈本来的操作，很多同学会考虑使用堆来维护最小值，但是这是不对的，因为堆每次找最小值的时间复杂度为\(O(log_2n)\)

题解：

可以考虑使用两个栈来共同维护，一个栈\(S\)用来进行元素常规的出栈和入栈操作，另一个栈\(Min\)维护最小值，其中栈顶元素表示当前栈中的最小值。对于入栈操作，首先将\(X\)入栈\(S\)，如果\(X\)比\(Min\)中的栈顶元素小，或者\(Min\)为空，则将\(X\)加入\(Min\)中。对于\(min\)操作，直接取栈\(Min\)中的栈顶元素。对于出栈操作，栈\(S\)直接出栈，如果当前栈顶元素等于\(Min\)的栈顶元素，则把\(Min\)的栈顶出栈

#include "bits/stdc++.h"
using namespace std;
stack<int>s;
stack<int>Min;
void Push(int x){
    s.push(x);
    if(Min.empty()) Min.push(x);
    else{
        int tmp=Min.top();
        if(x&lt;=tmp) Min.push(x);
    }
}
int GetMin(){
    return Min.top();
}
int Pop(){
    int tmp=s.top(); s.pop();
    if(tmp==Min.top()&amp;&amp;!Min.empty()) Min.pop();
    return tmp;
}
int main()
{
    //freopen("in.txt","r",stdin);
    int t;
    scanf("%d",&amp;t);
    while(t--){
        int op;
        scanf("%d",&amp;op);
        if(op==0){
            printf("%d\n",GetMin());
        }else if(op==1){
            int x; scanf("%d",&amp;x);
            Push(x);
        }else{
            printf("%d\n",Pop());
        }
    }
    return 0;
}