思路
两个栈来实现队列
一个栈用来添加元素一个栈用来删除元素
只要添加的那个栈不为空,且删除的那个栈为空,就一直往第二个栈取添加
代码
import java.util.Stack;
/**
* 两个栈来实现队列
* 一个栈用来添加元素一个栈用来删除元素
* 只要添加的那个栈不为空,且删除的那个栈为空,就一直往第二个栈取添加
*/
public class MyQueue {
Stack<Object> stack1;
Stack<Object> stack2;
public MyQueue() {
stack1 = new Stack<>();
stack2 = new Stack<>();
}
public void push(Object node) {
stack1.push(node);
}
public Object pop() {
if (isEmpty()) {
return -1;
}
stack1ToStack2();
return stack2.pop();
}
public Object peek() {
if (isEmpty()) {
return -1;
}
stack1ToStack2();
return stack2.peek();
}
/**
* 栈1往栈2放元素的条件
* 只要添加的那个栈不为空,且删除的那个栈为空,就一直往第二个栈取添加
*/
public void stack1ToStack2() {
if (stack2.isEmpty()) {
while (!stack1.isEmpty()) {
stack2.push(stack1.pop());
}
}
}
public boolean isEmpty() {
return stack1.isEmpty() && stack2.isEmpty();//两个栈都为空,则队列为空
}
public static void main(String[] args) {
MyQueue queue = new MyQueue();
queue.push(1);
queue.push(2);
queue.push(3);
queue.pop();
System.out.println(queue.stack2);//[3, 2]
queue.pop();
System.out.println(queue.stack2);//[3]
}
}