面試題09. 用兩個棧實現隊列【LeetCode劍指offer】

題目:

面試題09. 用兩個棧實現隊列

在這裏插入圖片描述

思路

  1. 兩個棧,一個size(隊列中數據的數量)
  2. 入隊列直接壓入棧1,並且size++;
  3. 出隊列時,先判斷size是否爲0,若size 爲 0,則說明隊列空,返回-1;
  4. 否則直接取出棧2的棧頂元素,若棧2爲空,則把棧1 的所有元素一次性出棧壓入棧2,此時棧2,按順序彈出則是隊列出隊列的順序,先進先出;出隊列時,要相應的的size–;

實現:

class CQueue {
    Stack<Integer> stack1;
    Stack<Integer> stack2;
    int size;


    public CQueue() {
        stack1 = new Stack();
        stack2 = new Stack();
        size = 0;
    }

    public void appendTail(int value) {
        stack1.push(value);
        size++;
    }

    public int deleteHead() {
        if (size == 0) {
            return -1;
        }
        if (stack2.empty()) {
            while (!stack1.empty()) {
                stack2.push(stack1.pop());
            }
        }
        size--;
        return stack2.pop();
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章