用两个栈实现队列
题目:用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
解题思路:
用一个栈只进,一个栈只出。进队列时只管压栈stack1,出队列时先判断之前压入stack2的数据是否全部弹出,不为空则只需出栈一个即可,为空则需要从stack1全部弹出压入stack2,然后stack2再执行一次出栈。
注意:这样写已经过了牛客网测试,但是在pop时其实应该先判断两个栈是否都为空,也就是判断队列是否为空然后抛出异常满足健壮性,不过不写这一步栈为空也会报异常。
import java.util.Stack;
public class Solution {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) {
stack1.push(node);
}
public int pop() {
if(stack2.isEmpty()) {
while(!stack1.isEmpty()) {
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}