Leetcode 225. Implement Stack using Queues 用佇列實現棧

zhangzhetaojj發表於2018-06-27

題目:

使用佇列實現棧的下列操作:

  • push(x) -- 元素 x 入棧
  • pop() -- 移除棧頂元素
  • top() -- 獲取棧頂元素
  • empty() -- 返回棧是否為空

注意:

  • 你只能使用佇列的基本操作-- 也就是 push to backpeek/pop from frontsize, 和 is empty 這些操作是合法的。
  • 你所使用的語言也許不支援佇列。 你可以使用 list 或者 deque(雙端佇列)來模擬一個佇列 , 只要是標準的佇列操作即可。
  • 你可以假設所有操作都是有效的(例如, 對一個空的棧不會呼叫 pop 或者 top 操作)。

解題思路:

使用一個佇列模擬棧的功能,棧和佇列的區別其實就是資料訪問的順序不同,把佇列調整成棧的資料訪問順序即可模擬棧的功能。

push方法修改資料訪問順序:

1
1 2 -> 2 1
2 1 3 -> 3 2 1
3 2 1 4 -> 4 3 2 1

程式碼實現:

class MyStack {

    Queue<Integer> queue;

    /** Initialize your data structure here. */
    public MyStack() {
        queue = new LinkedList<Integer>();
    }

    /** Push element x onto stack. */
    public void push(int x) {
        queue.offer(x);
        for (int i = 0; i < queue.size() - 1; i ++) {
            queue.offer(queue.poll());
        }
    }

    /** Removes the element on top of the stack and returns that element. */
    public int pop() {
        if (!queue.isEmpty()) return queue.poll();
        else return 0;
    }

    /** Get the top element. */
    public int top() {
        if (!queue.isEmpty()) return queue.peek();
        else return 0;
    }

    /** Returns whether the stack is empty. */
    public boolean empty() {
        return queue.isEmpty();
    }
}

/**
 * Your MyStack object will be instantiated and called as such:
 * MyStack obj = new MyStack();
 * obj.push(x);
 * int param_2 = obj.pop();
 * int param_3 = obj.top();
 * boolean param_4 = obj.empty();
 */

相關文章