建立一个Stack类,有min、push、pop功能,要在O(1)时间实现这三个功能
经典面试/笔试题目:(这里是同学hncujing的代码,比较简单我就不再写了,只标注下重点)
说下想法:简历两个stack容器,第一个存储栈应存的数据,第二个只存储最小值。
#include<iostream>
#include<string>
#include<stack>
using namespace std;
struct StackGetMin{
stack<int> elements,minStack;
void push(int x)
{
elements.push(x);
if(minStack.empty()||x<=minStack.top())
minStack.push(x);
}
bool pop()
{
if(elements.empty()) return false;
if(elements.top()==minStack.top())
minStack.pop();
elements.pop();
return true;
}
bool getMin(int &min) //这里要注意,如果栈为空返回false,bool型的函数无法返回最小值,所以用一个int型引用来处理
{
if(minStack.empty())
return false;
else
{
min=minStack.top();
return true;
}
}
};
int main()
{
StackGetMin stackMin;
stackMin.push(2);
stackMin.push(6);
stackMin.push(4);
int min;
stackMin.getMin(min);
cout<<min<<endl;
return 0;
}