如何用两个栈来实现一个队列,完成队列的Push和Pop操作(队列中的元素为int类型)
import java.util.Stack;
public class Solution {
//创建两个栈stack1、stack2
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
//入队
public void push(int node) {
stack1.push(node);//设定每一次入队都放入stack1中
}
//出队
public int pop() {
int num = 0;//定义一个num用于存放出队的数
if (stack1.empty()) return 0;
while (stack1.size() != 1) {
//将stack1中除了栈底元素之外元素放入stack2中
stack2.push(stack1.pop());
}
num = stack1.pop();//将栈底元素弹出
while(!stack2.empty()) {
//将stack2中所有元素放回到stack1中
stack1.push(stack2.pop());
}
return num;
}
}
思路:
- 有两个栈,设置其中一个栈stack1为专门存放数字的,另一个栈stack2作为过渡栈;
- 每次入队时都往stack1中放入,出队时先将stack1中除了栈底最后一个元素都先将其放入stack2中(栈是先进后出,队列是先进先出);
- 执行完该次循环后(stack1中只剩最后一个栈底元素),将栈底元素弹出并保存,但不要忘了再次利用循环将stack2中移入的元素再次弹出到stack1中,这样就确保了元素的顺序;
- 最后返回保存了的栈底元素值即可。