用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
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()) {
return stack2.pop();
} else if (!stack1.isEmpty() && stack2.isEmpty()) {
while (stack1.size() > 0)
stack2.push(stack1.pop());
return stack2.pop();
} else {
return 0;
}
}
}
这道题给了我们一个很好的思想,假如题目直说让你用栈来实现队列你是否会想到用到两个栈呢!是不是很巧妙,哈哈。就是还有一点需要转过弯,我们借用stack2来实现出队列,当需要加入新的数据时直接加入stack1中,而这个时候如果需要出队列stack2不为空,还是从stack2中出栈,这样是完全没有问题的!