最近一直在复习一些算法及数据结构方面的东西,就找了一个适合找工作笔试的题目,在剑指Offer上刷了几道题目,发现对复习知识点还是很有用的,做到重建二叉树这块。递归传值出了点问题,debug半小时才找出错误,所有还是写篇博客记录一下。也推荐要找工作的伙伴去剑指Offer刷题。
题目描述
用两个栈来实现一个队列,完成队列的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.size() > 0) {
return stack2.pop();
}else {
while(stack1.size() > 0 ) {
stack2.push(stack1.pop());
}
return stack2.pop();
}
}
}
很简单,一个栈专门负责进队的操作,假如要出队的话,就将负责进队的这个栈全部出栈到另外一个栈,假如又有进队的元素,直接进入stack1,等stack2的长度为0之后才将stack1出栈到stack2