题目:
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
思路: 使用vector实现栈的push和pop功能,额外添加一个ptr指针始终指向栈的最小值;
class MinStack {
private:
vector<int> data;
int ptr;
public:
/** initialize your data structure here. */
MinStack() {
ptr = 0;
}
void push(int x) {
data.push_back(x);
if(x < data[ptr]){
ptr = data.size() - 1;
}
}
void pop() {
if(ptr == data.size() - 1){
ptr = 0;
for(int i = 1; i < data.size() - 1;i++)
{
if(data[i] <= data[ptr]){
ptr = i;
}
}
}
data.pop_back();
}
int top() {
return data[data.size() - 1];
}
int getMin() {
return data[ptr];
}
};
/**
* 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();
*/