剑指Offer_7_用两个栈实现一个队列

2017-03-21

题目描述

使用两个栈实现一个队列的push和pop方法

解题思路

入栈时直接将数据存储到s1中，栈顶为最新元素；
出栈时查看s2中是否有数据，如果没有，则将s1中的数据弹出并压入s2，弹出s2栈顶元素为最旧元素，弹出即可；
当s1将所有数据压入s2后，s2仍然为空，则表示队列空；

代码实现

public class Solution
{
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    public void push(int x)
    {
        stack1.push(x);
    }
    public int pop()
    {
        if (stack2.isEmpty())
        {
            while (stack1.size() > 0)
                stack2.push(stack1.pop());
        }
        if (stack2.isEmpty())
        {
            try
            {
                throw new Exception("没有数据了");
            } catch (Exception e)
            {
                e.printStackTrace();
            }
        }
        return stack2.pop();
    }
    public static void main(String[] args)
    {
        Solution solution = new Solution();
        for (int i = 0; i < 5; i++)
        {
            solution.push(i);
        }
        System.out.println(solution.pop());
        System.out.println(solution.pop());
        System.out.println(solution.pop());
        System.out.println(solution.pop());
        System.out.println(solution.pop());
        System.out.println(solution.pop());
    }
}

补充

使用两个队列实现一个栈。解法类似。