剑指offer题目:用两个栈实现队列

国庆七天乐,刷个算法恢复一下“功力”。剑指Offer中遇到一题:

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

想想应该比较容易,两个栈实现队列,无非就是通过两个栈“倒腾”一下,实现先进先出的功能。

我们可以这样假设,每当需要进入队列的时候,我们都把它压入stack1,当需要弹出的时候,我们把stack1中的元素弹出并压如stack2,此时stack2的栈顶就是队列中最先进入的元素,pop即可,代码也很简单:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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.empty()){
while (!stack1.empty()) {
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}

比较基础的一道题,顺利通过检测。

0%