剑指 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; } };
这个题目想通这一点是一点都不难,可惜可惜,感觉自己不应该做不出来,唉。。。
浙公网安备 33010602011771号