题目:
正如标题所述,你需要使用两个栈来实现队列的一些操作。
队列应支持push(element),pop() 和 top(),其中pop是弹出队列中的第一个(最前面的)元素
pop和top方法都应该返回第一个元素的值。
样例:
比如push(1), pop(), push(2), push(3), top(), pop(),你应该返回1,2和2
挑战:
仅使用两个栈来实现它,不使用任何其他数据结构,push,pop 和 top的复杂度
都应该是均摊O(1)的
思路:
把栈2中的元素导入到栈1中就从先进后出队列变成了先进先出队列
代码:
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>();
}
private void stack2ToStack1(){
while(!stack2.isEmpty()){
stack1.push(stack2.pop());
}
}
public void push(int element) {
// write your code here
stack2.push(element);
}
public int pop() {
// 栈1为空把栈2元素加入栈1,栈1不为空直接从栈1抛出元素
if(stack1.empty() == true){
this.stack2ToStack1();
}
return stack1.pop();
}
public int top() {
if(stack1.empty() == true){
this.stack2ToStack1();
}
return stack1.peek();
}
}
代码可以优化:
http://www.cnblogs.com/wanghui9072229/archive/2011/11/22/2259391.html