剑指offer-包含min函数的栈(C++)

1、题目描述

定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。

2、解题思路
定义两个栈stack1和min_stack,stack1用来存储数据,min_stack用来记录stack1的最小值,假设入栈序列为:4、5、6、3、2、1,那么min_stack各轮次对应的min值就是:4、4、4、3、2、1。这样min_stack的栈顶的值就是stack1中数据的最小值。出栈时把两个栈中的元素同时弹出即可。
代码如下:

class Solution {
public:
    void push(int value) {
        stack1_.push(value);
        if(min_stack_.empty())
            min_stack_.push(value);
        else if (min_stack_.top() < value)
            min_stack_.push(min_stack_.top());
        else
            min_stack_.push(value);
    }
    void pop() {
        if(!stack1_.empty() && !min_stack_.empty())
        {
            stack1_.pop();
            min_stack_.pop();
        }
    }
    int top() {
        return stack1_.top();
    }
    int min() {
        return min_stack_.top();
    }
    
private:
    stack<int> stack1_;//主要存数据的栈
    stack<int> min_stack_;//存最小值的栈
};
原创文章 62 获赞 133 访问量 2万+

猜你喜欢

转载自blog.csdn.net/weixin_43142797/article/details/104960625