题目
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路
本来想的很简单,从stack1往stack2中一倒就把“后进先出”倒成了“先进先出”。
但是这个思路有一个问题,就是stack2里面有元素的时候。本来他应该先出的,反倒在stack2中先进,变成了后出
所以应该把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) {
this.stack1.push(node);
}
public int pop() {
if(stack2.isEmpty()){
while(!this.stack1.isEmpty()){
this.stack2.push(stack1.pop());
}
return stack2.pop();
}else{
stack2.pop();
}
}
}