题目:
实现一个特殊的栈,在实现栈的基本功能的基础上,再实现返回栈中最小元素的操作。
要求:
1、pop,push,getMin的操作时间复杂度都是O(1).
2、设计的栈类型可以使用现成的栈结构。
设置两个栈,一个data栈用来压数据,另一个min栈用来同步的压当前元素中的最小值,当data栈压入一个数,min栈就比较当前数与min栈栈顶的元素,谁小压谁
#include<stack>
#include <iostream>
using namespace std;
class MyStack{
private:
stack<int>stackData;
stack<int>stackMin;
public:
void push(int newNum){
if(stackMin.empty()){
stackMin.push(newNum);
}
else if(newNum<this->getMin()){
stackMin.push(newNum);
}
else{
stackMin.push(stackMin.top());
}
stackData.push(newNum);
}
int pop(){
if(this->stackData.empty()){
cout<<"your stack is empty."<<endl;
}
this->stackMin.pop();
return this->stackData.top();
}
int getMin(){
if(this->stackMin.empty()){
cout<<"your stack is empty";
}
return this->stackMin.top();
}
};