题目描述:
正如标题所述,你需要使用两个栈来实现队列的一些操作。
队列应支持push(element),pop() 和 top(),其中pop是弹出队列中的第一个(最前面的)元素。
pop和top方法都应该返回第一个元素的值。
样例:
比如push(1), pop(), push(2), push(3), top(), pop(),你应该返回1,2和2
思路讲解:首先我的思路就是两个栈,一个栈是用来进的,然后通过这个栈将里面的内容再放到另外一个栈中,这样就得到了顺序的结果,这样在队列出的时候就可以用这个栈,起栈顶元素就是应该出队列的元素。图示如下:
但是按照上面的,我做了之后,发现超时,分析发现原因是我再出队列之后,有需要重新将第二栈中的元素更新到第一栈中,这样是为了下一次有新的元素进队列,这样就导致我们花费大量的时间重复栈中不是栈顶元素的进栈与出栈,导致时间超时,后面我发现我们不需要将两个栈中的元素同步更新,以为队列也是有先后之分的,不会存在插队的,所以我们只需要每次将第二个栈中的元素pop出来,如果第二个中没有元素了,说明前面从第一个栈中进栈的元素,全部出栈了即出队列了,这次我们就需要将第一个栈中的新进的元素,再次加载进第二个栈中。top的操作也是一样的。只是每次不需要pop出去。
代码详解:
class MyQueue {
public:
MyQueue() {
// do intialization if necessary
}
/*
* @param element: An integer
* @return: nothing
*/
void push(int element) {
// write your code here
pos_stack.push(element);
}
/*
* @return: An integer
*/
int pop() {
// write your code here
if(neg_stack.empty()){//判断第二个栈中是否还有元素,由则直接弹出,没有就要将第一个栈中的元素放入第二个栈中
while(!pos_stack.empty()){
neg_stack.push(pos_stack.top());
pos_stack.pop();
}
}
int res=neg_stack.top();
neg_stack.pop();
return res;
}
/*
* @return: An integer
*/
int top() {
if(neg_stack.empty()){
while(!pos_stack.empty()){
neg_stack.push(pos_stack.top());
pos_stack.pop();
}
}
int res=neg_stack.top();
return res;
}
private:
stack<int>pos_stack;
stack<int>neg_stack;
};