JZ5 使用两个栈实现队列

描述

用两个栈来实现一个队列，分别完成在队列尾部插入整数(push)和在队列头部删除整数(pop)的功能。 队列中的元素为int类型。保证操作合法，即保证pop操作时队列内已有元素。

解析:

"PSH1":代表将1插入队列尾部

"PSH2":代表将2插入队列尾部

"POP“:代表删除一个元素，先进先出=>返回1

"POP“:代表删除一个元素，先进先出=>返回2

示例1

输入：["PSH1","PSH2","POP","POP"]

返回值：1,2

 

1. 分析

队列的特性是：“先入先出”，栈的特性是：“先入后出”

当我们向模拟的队列插入数 a,b,c 时，假设插入的是 stack1，此时的栈情况为：

• 栈 stack1：{a,b,c}
• 栈 stack2：{}

当需要弹出一个数，根据队列的"先进先出"原则，a 先进入，则 a 应该先弹出。但是此时 a 在 stack1 的最下面，将 stack1 中全部元素逐个弹出压入 stack2，现在可以正确的从 stack2 中弹出 a，此时的栈情况为：

• 栈 stack1：{}
• 栈 stack2：{c,b}

继续弹出一个数，b 比 c 先进入"队列"，b 弹出，注意此时 b 在 stack2 的栈顶，可直接弹出，此时的栈情况为：

• 栈 stack1：{}
• 栈 stack2：{c}

此时向模拟队列插入一个数 d，还是插入 stack1，此时的栈情况为：

• 栈 stack1：{d}
• 栈 stack2：{c}

弹出一个数，c 比 d 先进入，c 弹出，注意此时 c 在 stack2 的栈顶，可直接弹出，此时的栈情况为：

• 栈 stack1：{d}
• 栈 stack2：{c}

根据上述栗子可得出结论：

1. 当插入时，直接插入 stack1
2. 当弹出时，当 stack2 不为空，弹出 stack2 栈顶元素，如果 stack2 为空，将 stack1 中的全部数逐个出栈入栈 stack2，再弹出 stack2 栈顶元素

2. 代码

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18

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) {             while (stack1.size() != 0) {                 stack2.push(stack1.pop());             }         }         return stack2.pop();     } }

3. 复杂度

push时间复杂度：O(1)
pop空间复杂度：O(1)