剑指offer-9 用两个栈实现队列
微信搜索【程序员画工师】关注更多Java编程技术、数据结构与算法、面试题相关内容。
题目
用两个栈来实现一个队列,完成队列的Push和Pop操作。队列中的元素为int类型。
思路
这道题的意图是要求我们用两个"先进后出"的栈实现一个“先进先出”的队列。分析该队列插入和删除元素的过程:两个栈stack1、stack2,插入元素的时候,把它先插入stack1,假设插入{a,b,c};当需要删除元素的时候,队列是先入先出的原则,应该出队的是元素a,元素a在stack1的栈底,并不在栈顶上,所以不能直接删除。stack2是空的,此时就可以把stack2利用起来。那么我们把stack1中的元素逐个弹出并压入stack2,这样的话,stack1为空,stack2中的元素为{c,b,a},需要出队的元素a位于栈顶,就可以直接弹出了。所以删除元素的步骤为:stack2为空的时候,将stack1的元素逐个弹出压入stack2;stack2不为空的时候,栈顶元素是最先进入队列的,可以直接弹出。当再新入队一个元素d的时候,把它压入stack1,下次删除的时候如果stack2不为空,直接弹出栈顶元素,而stack2中的栈顶元素确实比d先进入队列,因此不会有问题。
时间复杂度分析:push()操作,时间复杂度为o(1)
pop()操作,如果主栈空,需要将元素从辅助栈弹出并压入主栈,复杂度为o(n) ;如果主栈不空,只需要从主栈弹出元素,复杂度为 o(1) 。
上代码
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.isEmpty()){
while(!stack1.isEmpty()){
stack2.push(stack1.pop());
}
}
if(stack2.isEmpty()){
throw null;
}
return stack2.pop();
}
}
References
[1] 《剑指offer(第二版)》 何海涛著
程序员画工师公众号,欢迎交流