实现包含min函数的栈，使得min时间复杂度为O(1)

虚竹777777

于 2018-10-29 09:37:11 发布

阅读量258

点赞数

CC 4.0 BY-SA版权

分类专栏：算法练习

本文链接：https://blog.youkuaiyun.com/baidu_38370610/article/details/83501233

算法练习专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一种使用两个栈实现的高效最小值查找算法。通过同步维护数据栈和最小值栈，该算法能在O(1)时间内获取当前栈中最小元素，同时支持元素的压入和弹出操作。此方法在数据结构和算法设计中具有广泛应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#include <cstring>
#include <stack>
using namespace std;
stack<int> a_stack;
stack<int> min_stack;
void push(int n)
{
    a_stack.push(n);
    if(min_stack.size()==0) min_stack.push(n);
    else
    {
        int top = min_stack.top();
        if(n<top) min_stack.push(n);
        else min_stack.push(top);
    }
}
int minEle()
{
    if(min_stack.size()>0) return min_stack.top();
    else return -1;
}
int pop()
{
    if(a_stack.size()>0 && min_stack.size()>0)
    {
        int n = a_stack.top();
        a_stack.pop();
        min_stack.pop();
        return n;
    }
    else return -1;
}
int main()
{
    int n,val;
    for(;;)
    {
        cin>>n;
        if(n==1)
        {
            cin>>val;
            push(val);
        }
        else if(n==2)
        {
            cout<<minEle()<<endl;
        }
        else cout<<pop()<<endl;
    }
    return 0;
}