面试题(判断元素出栈入栈顺序的合法性。)
面试题(两个栈实现一个队列和两个队列实现一个栈)
实现一个栈,要求实现Push(出栈)、Pop(入栈)、Min(返回最小值的操作)的时间复杂度为O(1)
解题思路:
我们定义两个栈分别为s1,s2。我们用s2保存最小值,s1保存原来的值
Push():我们添加元素到s1中,如果s2为空或者x
#include<iostream>
#include<stack>
using namespace std;
template<class T>
class Stack
{
public:
void Push(const T&x)
{
s1.push(x);
if (s2.empty() || x < s2.top())
{
s2.push(x);
}
else
{
s2.push(s2.top());
}
}
void Pop()
{
s1.pop();
s2.pop();
}
T GetMin()
{
if (!s2.empty())
{
return s2.top();
}
}
protected:
stack<int> s1;
stack<int> s2;
};
void Test1()
{
Stack<int> l;
l.Push(5);
l.Push(3);
l.Push(2);
l.Push(6);
cout << l.GetMin() << endl;
l.Pop();
l.Pop();
cout << l.GetMin() << endl;
}