剑指offer——包含min函数的栈

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/cherish0222/article/details/79368729

题目描述:定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。

分析:定义一个辅助栈,保存对应元素存在时当前栈中的最小元素,即可在O(1)的时间内找到最小的数。

参考代码:

class Solution {
public:
    void push(int value) {
        if(s_data.empty()){
            s_data.push(value);
            s_min.push(value);
        }
        else{
            s_min.push(value<s_min.top()?value:s_min.top());
            s_data.push(value);
        }
    }
    void pop() {
        s_data.pop();
        s_min.pop();
    }
    int top() {
        return s_data.top();
    }
    int min() {
        return s_min.top();
    }
private:
    stack<int> s_data;
    stack<int> s_min;
};

阅读更多

没有更多推荐了,返回首页