剑指offer(9 用两个栈实现队列) 题解

剑指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(第二版)》 何海涛著

程序员画工师公众号，欢迎交流