【剑指offer】用栈实现队列

时间:2022-05-11 17:40:54

描述:

正如标题所述,你需要使用两个栈来实现队列的一些操作。

队列应支持push(element),pop() 和 top(),其中pop是弹出队列中的第一个(最前面的)元素。

pop和top方法都应该返回第一个元素的值。

样例:

比如push(1), pop(), push(2), push(3), top(), pop(),你应该返回1,2和2


Java代码:

public class MyQueue {
private Stack<Integer> stack1;
private Stack<Integer> stack2;

public MyQueue() {
// do initialization if necessary
stack1 = new Stack<Integer>();
stack2 = new Stack<Integer>();
}

public void push(int element) {
stack1.add(element);
}

//弹出第一个数,并返回值
public int pop() {
while(!stack1.isEmpty()){
stack2.add(stack1.pop());
}
int ret = stack2.pop();

while(!stack2.isEmpty()){
stack1.add(stack2.pop());
}

return ret;
}

//返回第一数值,不弹出
public int top() {
while(!stack1.isEmpty()){
stack2.add(stack1.pop());
}
int ret = stack2.peek();

while(!stack2.isEmpty()){
stack1.add(stack2.pop());
}

return ret;
}
}