题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
Note:寻找stack1中所含最小元素时,用另一个栈stack2存放读入的数据
class Solution {
public:
void push(int value) {
stack1.push(value);
}
void pop() {
stack1.pop();
}
int top() {
return stack1.top();
}
int min() {
int minNum=0;
if(!stack1.empty()){
minNum=stack1.top();
stack2.push(stack1.top());
stack1.pop();
}
while(!stack1.empty()){
if(minNum>stack1.top()){
minNum=stack1.top();
stack2.push(stack1.top());
stack1.pop();
}
else{
stack2.push(stack1.top());
stack1.pop();
}
}
while(!stack2.empty()){
stack1.push(stack2.top());
stack2.pop();
}
return minNum;
}
private:
stack<int> stack1;
stack<int> stack2;
};