题目描述:
使用栈实现队列的下列操作:
push(x) – 将一个元素放入队列的尾部。
pop() – 从队列首部移除元素。
peek() – 返回队列首部的元素。
empty() – 返回队列是否为空。
示例:
MyQueue queue = new MyQueue();
queue.push(1);
queue.push(2);
queue.peek(); // 返回 1
queue.pop(); // 返回 1
queue.empty(); // 返回 false
说明:
你只能使用标准的栈操作 – 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
假设所有操作都是有效的 (例如,一个空的队列不会调用 pop 或者 peek 操作)。
方法1:
主要思路:
(1)直观的想,使用两个栈实现队列,一个栈用来直接接受压入的数据s_push,另一个队列用来弹出数据s_pop,当弹出的栈s_pop为空时,将压入的栈s_push中的数据全部弹出到s_pop中,这样就实现了数据的反序,既从栈的先入后出,变为队列的先入先出;
class MyQueue {
public:
stack<int> s_push;
stack<int> s_pop;
/** Initialize your data structure here. */
MyQueue() {
}
/** Push element x to the back of queue. */
void push(int x) {
s_push.push(x);//接受数据
}
/** Removes the element from in front of queue and returns that element. */
int pop() {
if(s_pop.empty()){//若弹出数据的栈为空,则将压入数据的栈中的数据转移到弹出栈中
while(!s_push.empty()){
s_pop.push(s_push.top());
s_push.pop();
}
}
int tmp=s_pop.top();//正常的要弹出的数据
s_pop.pop();
return tmp;
}
/** Get the front element. */
int peek() {
if(s_pop.empty()){//和弹出数据一个道理
while(!s_push.empty()){
s_pop.push(s_push.top());
s_push.pop();
}
}
return s_pop.top();
}
/** Returns whether the queue is empty. */
bool empty() {
return s_pop.empty()&&s_push.empty();//只有两个栈都为空时,才能说明现在没有数据存在
}
};
/**
* Your MyQueue object will be instantiated and called as such:
* MyQueue* obj = new MyQueue();
* obj->push(x);
* int param_2 = obj->pop();
* int param_3 = obj->peek();
* bool param_4 = obj->empty();
*/