题目:
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
栈的特点是什么?
队列的特点是什么?
- 存放数据:用其中的栈stack1去存放数据,这样就变成先进的数据在栈底的顺序存放在stack1中;
- 取数据:从stack1栈顶取数据存放到另一个栈stack2中,这样数据由正→反→正,再次变为先进的数据在栈顶的顺序,这时在使用stack2的pop()方法,取栈顶数据,就实现了先进先出的特点;
这种能实现的前提是,每次都是所有数据出栈完,再存放数据;
如果出现出栈一个数据后,又存放数据的情况,就会和我们想要的结果不一样;
那怎么办呢?
问题不大,只要出栈时,先校验存放正序的栈stack2是否为空,如果不为空,先取stack2栈顶的数据即可;
源码如下:
public class LinkNode {
public Stack<Integer> stack1 = new Stack<>();
public Stack<Integer> stack2 = new Stack<>();
public void push(int node){
stack1.push(node);
}
public int pop(){
if(!stack2.isEmpty()){
return stack2.pop();
}
while(!stack1.isEmpty()){
stack2.push(stack1.pop());
}
return stack2.pop();
}
}