劍指offer-兩個棧實現-隊列尾部插入,頭部刪除

大家都知道,隊列是一個尾部(rear)插入,頭部(front)刪除的數據結構。本題要求,用兩個棧,構造出一個隊列出來。

本題中,構造兩個棧,stack1和stack2,1用來插入,2用來彈出。其中,棧1的插入很簡單,函數體內部,直接用add方法即可。但是棧2的彈出,要考慮此時棧2是不是有值,如果2爲空,1不空。則彈出1的值,add進2。如果此時2還沒有值,則拋異常,如果有值,則pop出來。

拓:只考慮了2的情況,沒考慮1的情況,要是1滿了,壓不進去呢,這在後面還可以商榷一下。

import java.util.Stack;

public class Test07 {
    /**
     * 用兩個棧模擬的隊列
     * 用兩個棧實現一個隊列。隊列的聲明如下,諸實現它的兩個函數appendTail和deleteHead,
     * 分別完成在隊列尾部插入結點和在隊列頭部刪除結點的功能。
     */
    public static class MList<T> {
        // 插入棧,只用於插入的數據
        private Stack<T> stack1 = new Stack<>();
        // 彈出棧,只用於彈出數據
        private Stack<T> stack2 = new Stack<>();

        public MList() {
        }
        
        // 添加操作,在隊列尾部插入結點
        public void appendTail(T t) {
            stack1.add(t);
        }

        // 刪除操作,在隊列頭部刪除結點
        public T deleteHead() {

            // 先判斷彈出棧是否爲空,如果爲空就將插入棧的所有數據彈出棧,
            // 並且將彈出的數據壓入彈出棧中
            if (stack2.isEmpty()) {
                while (!stack1.isEmpty()) {
                    stack2.add(stack1.pop());
                }
            }

            // 如果彈出棧中還沒有數據就拋出異常
            if (stack2.isEmpty()) {
                throw new RuntimeException("No more element.");
            }

            // 返回彈出棧的棧頂元素,對應的就是隊首元素。
            return stack2.pop();
        }
    }
}

參考文獻:github上某大神的代碼。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章