剑指offer5-用两个栈实现队列
最近全国疫情严重,待在家里没事干,马上又要准备春招了,最近刷刷题,记录一下!再说一句,武汉加油,大家出门记得戴口罩!
1、题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
2、思路分析
本题的基本意图是:用两个后入先出的栈来实现先入先出的队列。
对于这个问题,我们可以通过一个实例来进行具体分析。不难得出相应的规律:有两个栈stack1和stack2,每次向队列中插入元素可以都压入到stack1中,当需要从队列中删除元素时,我们应该是删除最早插入的那个(FIFO),这时可以将stack1中的元素逐个弹出并压入stack2,直到stack1为空,这时最早插入的元素就位于stack2的栈顶,可以直接弹出。
因此,我们总结如下:压入元素时,都压入栈1,当需要弹出时,从栈2弹出,当栈2不为空时直接弹出栈顶元素,为空时将栈1的元素“倒进去”。
3、代码
package com.jianzhioffer;
import java.util.Stack;
public class Main05 {
//栈 先进后出
//队列 先进先出
//push 操作在栈1里面完成 pop操作在栈2里面完成
/*
方法一:通过两个栈中元素之间的复制交换去实现队列
*/
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) {
while(!stack2.isEmpty()) {
stack1.push(stack2.pop()); //将栈2中的元素加入到栈1中
}
stack1.push(node);
}
public int pop() {
while(!stack1.isEmpty()) {
stack2.push(stack1.pop()); //将栈1中的元素加入到栈2中
}
return stack2.pop();
}
}