定义栈的数据结构并得到最小元素

本文介绍一种特殊的栈数据结构,在该结构中实现了O(1)时间复杂度的min函数,用于获取栈内的最小元素。通过使用两个栈,其中一个正常操作而另一个仅存放最小值的方法,确保了min、push及pop操作都能保持O(1)的时间复杂度。

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

题目描述:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)

栈的数据结构都是很简单的,此题的主要难点就是如何在时间复杂度为O(1)的情况下找到最小元素

思路:要想只调用一次函数就得到最小元素,那么最小元素必须是栈顶元素。要达到这个目的,只能是在元素入栈时就比较之后再存放,将最小的较小的元素放在栈顶(注意:这里说的是较小,而不是最小)。对于一个栈来说,随着入栈元素的增加,不一定能将最小的元素放在栈顶,如图:

 

所以我们需要定义两个栈,栈1正常入栈出栈,栈2只存放最小的元素,其它元素不存放,为了达到这个目的,我们只需将每一次要入栈的元素与栈2的栈顶元素进行比较,将较小的入栈即可(因为每次都要入栈较小的,所以可能栈2中的元素都是相同的,但是能保证是最小的)

代码实现:

class MinStack {
    stack<int>st;
    stack<int>minStack;
public:
    MinStack() {
        while(!st.empty())
        {
            st.pop();
        }
        while(!minStack.empty())
        {
            minStack.pop();
        }
        minStack.push(INT_MAX);
    }
    
    void push(int x) {
        st.push(x);
        int minVal=std::min(minStack.top(),x);
        minStack.push(minVal);
    }
    
    void pop() {
     st.pop();
     minStack.pop();
    }
    
    int top() {
        return st.top();
    }
    
    int min() {
        return minStack.top();
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值