232. Implement Queue using Stacks

一、题目

　　1、审题

　　

　　2、分析

　　　　使用栈来实现队列的功能。

 

二、解答

　　1、思路

　　　　方法一、

　　　　　　只修改 Pop 方法，使得栈中元素排序为队列的规则。

public class MyQueue {
    
    Stack<Integer> stack;
    /** Initialize your data structure here. */
    public MyQueue() {
        stack = new Stack<>();
    }
    
    /** Push element x to the back of queue. */
    public void push(int x) {
        if(stack.isEmpty()) {
            stack.push(x);
        }
        else {
            Stack<Integer> tmp = new Stack<>();
            while(!stack.isEmpty())
                tmp.push(stack.pop());
            stack.push(x);
            while(!tmp.isEmpty())
                stack.push(tmp.pop());
        }
    }
    
    /** Removes the element from in front of queue and returns that element. */
    public int pop() {
        return stack.pop();
    }
    
    /** Get the front element. */
    public int peek() {
        return stack.peek();
    }
    
    /** Returns whether the queue is empty. */
    public boolean empty() {
        return stack.isEmpty();
    }
}

 

　　方法二、

　　　　使用两个 Stack，input、output。

　　　　push： 压入 input 栈。

　　　　pop、peek： 若 output 不为空，则对 output 操作； 若 output 为空，则将 input 元素全部压入 output，在对 output 操作。

　　　　empty： 当 output、input 全为空时才返回 true。

class MyQueue {

    Stack<Integer> input = new Stack<>();
    Stack<Integer> output = new Stack<>();
    
    public void push(int x) {
        input.push(x);
    }
    
    public int pop() {
        peek();
        return output.pop();
    }
    
    public int peek() {
        if(output.isEmpty()) {
            while(!input.isEmpty())
                output.push(input.pop());
        }
        return output.peek();
    }
    
    public boolean empty() {
        return input.empty() && output.empty();
    }
}