面试题30. 包含min函数的栈

面试题30. 包含min函数的栈

在这里插入图片描述
解题思路:
本题难点在于调用 min、push 及 pop 的时间复杂度都是 O(1)。栈调用push和pop复杂度就是O(1),但是要实现min()函数需要遍历栈对比,时间复杂度是Q(N),则要实现O(1)可以定义两个栈,一个数据栈A和一个辅助栈B。

  1. 数据栈 AA : 栈 AA 用于存储所有元素,保证入栈 push() 函数、出栈 pop() 函数、获取栈顶 top() 函数的正常逻辑。
  2. 辅助栈 BB : 栈 BB 中存储栈 AA 中所有 非严格降序 的元素,则栈 AA 中的最小元素始终对应栈 BB 的栈顶元素,即 min() 函数只需返回栈 BB 的栈顶元素即可。
  3. 辅助栈BB的push()函数实现:判断BB是否为空,空则直接将元素入栈,非空拿当前元素和BB的栈顶元素比较小于等于入栈。
  4. 辅助栈BB的pop()函数实现:判断当前AA的即将出战的元素(栈顶元素)是否和BB的栈顶元素相同,相同则一起出栈,不同则BB元素不出栈。
class MinStack {
    stack<int> stack1;
    stack<int> stack2;
public:
    /** initialize your data structure here. */
    MinStack() {
        while(!stack1.empty())
        {
            stack1.pop();
        }
        while(!stack2.empty())
        {
            stack2.pop();
        }
    }   
    void push(int x) {
        stack1.push(x);
        if(stack2.empty())
            stack2.push(x);
        else if(x <= stack2.top())
            stack2.push(x);
    }   
    void pop() {
        if(stack1.top() == stack2.top())
            stack2.pop();
        stack1.pop();
    } 
    int top() {
        int num = stack1.top();
        return num;
    }
    int min() {
        if(stack2.empty())
        {
            return NULL;
        }
        int num = stack2.top();
        return num; 
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值