-
题目:
使用栈实现队列的下列操作:push(x) – 将一个元素放入队列的尾部。
pop() – 从队列首部移除元素。
peek() – 返回队列首部的元素。
empty() – 返回队列是否为空。
示例:
MyQueue queue = new MyQueue();
queue.push(1);
queue.push(2);
queue.peek(); // 返回 1
queue.pop(); // 返回 1
queue.empty(); // 返回 false
- 解题思路:
用双栈实现队列:初始化两个栈,其中栈s1作为队列,s2作为辅助栈
1.实现push(),若s1为空,直接插入;若s1不为空,将s1的所有元素push入s2中,在将x push 入s1中,最后,将s2中的所有元素push入s1中.
2.实现pop(),只用将s1栈顶元素出栈即可
3.实现top(),只用返回s1栈顶元素即可
4.实现empty(),只用判断s1是否为空即可。
代码实现(C++)
/** Initialize your data structure here. */
stack<int> s1,s2;
int temp;
MyQueue() {
while(!s1.empty())
s1.pop();
while(!s2.empty())
s2.pop();
}
/** Push element x to the back of queue. */
void push(int x) {
if(s1.empty())
s1.push(x);
else
{
while(!s1.empty())
{
temp = s1.top();
s1.pop();
s2.push(temp);
}
s1.push(x);
while(!s2.empty())
{
temp = s2.top();
s2.pop();
s1.push(temp);
}
}
}
/** Removes the element from in front of queue and returns that element. */
int pop() {
temp = s1.top();
s1.pop();
return temp;
}
/** Get the front element. */
int peek() {
return s1.top();
}
/** Returns whether the queue is empty. */
bool empty() {
return s1.empty();
}