剑指 Offer 30. 包含min函数的栈

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

  示例:

    MinStack minStack = new MinStack();
    minStack.push(-2);
    minStack.push(0);
    minStack.push(-3);
    minStack.min(); --> 返回 -3.
    minStack.pop();
    minStack.top(); --> 返回 0.
    minStack.min(); --> 返回 -2.

========================================================================

这个题目比较麻烦的就是返回最小值的时候要求时间复杂度是O(1),我一开始的思路是被固定了,只想着查询时间复杂度是O(1),那不就是hash,但是hash怎么用?

挠头苦想了半个小时,实在是不知道怎么做了,看了眼评论,说加一个变量记录当前的最小值,一语惊醒梦中人,自己就陷在一个错误的思路里了,唉

上代码:

struct stackNode {
    int value;
    int min;
};

class MinStack {
private:
    vector<stackNode> *stack = new vector<stackNode>();
public:
    /** initialize your data structure here. */
    MinStack() {

    }
    void push(int x) {
        stackNode s;
        s.value = x;
        if (stack->empty())
            s.min = x;
        else {
            int mV = stack->back().min;
            if (mV > x)
                s.min = x;
            else
                s.min = mV;
        }
        stack->push_back(s);
    }
    void pop() {
        stack->pop_back();
    }
    int top() {
        stackNode s;
        s = stack->back();
        return s.value;
    }
    int min() {
        stackNode s;
        s = stack->back();
        return s.min;
    }
};

这个题目想通这一点是一点都不难,可惜可惜,感觉自己不应该做不出来,唉。。。

posted on 2022-01-05 22:38  4小旧  阅读(28)  评论(0)    收藏  举报

导航