一、问题描述
使用两个栈实现一个队列
二、算法分析
栈是先进后出,因此两个可以模拟实现先进先出
三、算法设计
定义数据结构
Stack<Integer> stack1
Stack<Integer> stack2
对于push操作:元素入队列时,将其加入到stack1中
对于pop操作:元素出队列时,先判断stack2是否为空,若不为空,则从stack2中弹出一个元素;若为空,则将stack1中的所有元素弹出到stack2中,然后弹出一个元素
四、代码实现
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(stack1.isEmpty()&&stack2.isEmpty()){
System.out.println("Queue is empty.");
}
if(stack2.isEmpty()){
while(!stack1.isEmpty()){
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
public static void main(String[] args) {
Solution queue = new Solution();
queue.push(1);
queue.push(2);
queue.push(3);
System.out.println(queue.pop());
queue.push(4);
System.out.println(queue.pop());
System.out.println(queue.pop());
System.out.println(queue.pop());
}
}