设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
思路:建立一个数组,因为不知道数组大小,所以用了vector<int>。
class MinStack {
public:
vector<int> obj;
/** initialize your data structure here. */
MinStack() {
}
void push(int x) {
obj.push_back(x);
}
void pop() {
obj.pop_back();
}
int top() {
return obj[obj.size()-1];
}
int getMin() {
int min=obj[0];
for(int i=0;i<obj.size();i++)
{
if(min>obj[i])
min=obj[i];
}
return min;
}
};
/**
* Your MinStack object will be instantiated and called as such:
* MinStack* obj = new MinStack();
* obj->push(x);
* obj->pop();
* int param_3 = obj->top();
* int param_4 = obj->getMin();
*/
看了答案以后,用了两个栈,原来这个题的关键是getmin的时间复杂度位o(1)。
思路是,每次压入栈mystack一个数a时,也计算当a为栈顶时的最小值,并放入min_stack中,这两个栈保持同步。
代码: