版权声明:本文为博主原创文章,未经博主允许不得转载 https://blog.csdn.net/TTTZZZTTTZZZ/article/details/86608068
/**
* 用两个栈实现队列
* <p>
* (1)对于插入操作,栈与队列都是从队尾进行,因此一行代码就可以完成push()
* (2)对于弹出操作,队列先进先出从队头开始,而栈后进先出从队尾开始,
* 要想取到队头元素,就得需要第二个栈stack2的协助:弹出时将stack1的元素依次取出放到stack2中
* 此时stack2进行弹出的顺序就是整个队列的弹出顺序。而如果需要插入,放到stack1中即可。
* 总结:stack1负责插入,stack2负责弹出,如果stack2为空了,将stack1的元素依次弹出并存放到stack2中,之后对stack2进行弹出操作
*
* @author VicterTian
* @version V1.0
* @Date 2019/1/23
*/
public class P68_QueueWithTwoStacks {
public static void main(String[] args) {
MyQueue<Integer> queue = new MyQueue<>();
queue.push(1);
queue.push(2);
queue.push(3);
System.out.println("queue.pop() = " + queue.pop());
System.out.println("queue.pop() = " + queue.pop());
System.out.println("queue.pop() = " + queue.pop());
}
}
/**
* 用栈实现队列
*/
class MyQueue<T> {
private Stack<T> stack1 = new Stack<>();
private Stack<T> stack2 = new Stack<>();
public void push(T data) {
stack1.push(data);
}
public T pop() {
if (!stack2.isEmpty()) {
return stack2.pop();
}
if (!stack1.isEmpty()) {
while (!stack1.isEmpty()) {
stack2.push(stack1.pop());
}
return stack2.pop();
}
return null;
}
}