正如标题所述,你需要使用两个栈来实现队列的一些操作。
队列应支持push(element),pop() 和 top(),其中pop是弹出队列中的第一个(最前面的)元素。
pop和top方法都应该返回第一个元素的值。
样例
比如push(1), pop(), push(2), push(3), top(), pop(),你应该返回1,2和2
public class MyQueue {
Stack stack1 = new Stack();//进来的元素都放在里面
Stack stack2 = new Stack();//元素都从这里出去
public MyQueue() {
// do intialization if necessary
}
/*
* @param element: An integer
* @return: nothing
*/
public void push(int element) {
// write your code here
stack1.push(element);
}
/*
* @return: An integer
*/
public int pop() {
// write your code here
if (stack2.isEmpty()) {//如果没有元素,就把负责放入元素的栈中元素全部放进来
while (!stack1.isEmpty()) {
stack2.push(stack1.pop());
}
}
return (int)stack2.pop();//有元素后就可以弹出了
}
/*
* @return: An integer
*/
public int top() {
// write your code here
if (stack2.isEmpty()) {
while (!stack1.isEmpty()) {
stack2.push(stack1.pop());
}
}
return (int)stack2.peek();
}
}