剑指offer:面试题22——包含min函数的栈

剑指offer:面试题22——包含min函数的栈

题目:

定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

思路:

使用两个栈,一个存放数据,另一个存放当前的最小值,但是如果最小值被弹出,还需要保存次小值,因此最小值栈的设计应该为:如果当前最小值栈为空,直接入栈;如果入栈数据小于当前最小值栈顶,直接入栈;如果当前元素大于最小值栈顶,则直接入栈当前最小值栈顶(即为当前最小值)

这样做的目的使得数据栈中每个元素在入栈时,当前栈中最小值得以保存

代码如下:

class Solution {
public:
    stack<int>datastack;   
    stack<int>minstack;
public:
    void push(int value) {
        datastack.push(value);
        if(minstack.empty())
            minstack.push(value);
        else
        {
            if(minstack.top()>value)
                minstack.push(value);
            else
                minstack.push(minstack.top());
        }
    }
    void pop() {
        if(!datastack.empty() && !minstack.empty())
            datastack.pop();
            minstack.pop();
    }
    int top() {
        return datastack.top();
    }
    int min() {
        return minstack.top();
    }
};

猜你喜欢

转载自blog.csdn.net/MereX/article/details/89634550