用栈实现队列
需要完成队列的push,pop,peek取队列的第一个元素值,empty。
【不实现】每个操作均摊时间复杂度为O(1)的简单方法的思路如下:
需要用两个栈,当需要实现队列的push操作时,直接向s1压入元素,时间复杂度为O(1)。
要实现pop操作时,需要把s1的元素压入s2中,一直到s1只剩一个元素为止,并将该元素出栈不压入s2,最后将s2中的元素压回s1,时间复杂度为O(n)。
要实现peek操作(取栈顶元素),同样操作,只是不将s1的最后一个元素出栈而是返回它的值,最后还是将s2中的元素压回s1,时间复杂度为O(n)。
empty操作,只需要检查s1是否为空,时间复杂度为O(1)。
进阶:实现队列每个操作均摊时间复杂度为O(1)。
同样使用两个栈,s1专门负责push操作。
实现pop操作,当s2为空的时候,将s1中的元素全部压入栈中,然后直接对s2出栈,此时的出栈操作时间复杂度为O(1),也就是说只有当s2为空的时候执行pop操作才需要O(n)的时间复杂度。
peek操作同理。
empty操作判断s1和s2是否为空,都是O(1)的时间复杂度。
进阶的代码如下:
class MyQueue {
public:
MyQueue() {
}
void push(int x) {
s1.push(x);
}
int pop() {
if(s2.empty())
{
// s1不为空,s2为空,此时将s1的所有元素出栈并压入s2中,完成逆序
s1Tos2();
}
int val = s2.top();
s2.pop();
return val;
}
int peek() {
if(s2.empty())
{
// s1不为空,s2为空,此时将s1的所有元素出栈并压入s2中,完成逆序
s1Tos2();
}
return s2.top();
}
bool empty() {
if(s2.empty() && s1.empty())
return true;
else
return false;
}
void s1Tos2() // s1不为空,s2为空,此时将s1的所有元素出栈并压入s2中,完成逆序
{
while(!s1.empty())
{
int temp = s1.top();
s1.pop();
s2.push(temp);
}
}
public:
stack<int> s1;
stack<int> s2;
};