死磕 java集合之TreeMap源碼分析（四）——紅黑樹全解析

時間 2019-11-24

標籤 java 集合 treemap 源碼分析解析欄目 Java 简体版

原文原文鏈接

歡迎關注個人公衆號「彤哥讀源碼」，查看更多源碼系列文章, 與彤哥一塊兒暢遊源碼的海洋。java

二叉樹的遍歷

咱們知道二叉查找樹的遍歷有前序遍歷、中序遍歷、後序遍歷。node

（1）前序遍歷，先遍歷我，再遍歷個人左子節點，最後遍歷個人右子節點；git

（2）中序遍歷，先遍歷個人左子節點，再遍歷我，最後遍歷個人右子節點；github

（3）後序遍歷，先遍歷個人左子節點，再遍歷個人右子節點，最後遍歷我；微信

這裏的前中後都是以「我」的順序爲準的，我在前就是前序遍歷，我在中就是中序遍歷，我在後就是後序遍歷。ide

下面讓咱們看看經典的中序遍歷是怎麼實現的：學習

public class TreeMapTest {

    public static void main(String[] args) {
        // 構建一顆10個元素的樹
        TreeNode<Integer> node = new TreeNode<>(1, null).insert(2)
                .insert(6).insert(3).insert(5).insert(9)
                .insert(7).insert(8).insert(4).insert(10);

        // 中序遍歷，打印結果爲1到10的順序
        node.root().inOrderTraverse();
    }
}

/** * 樹節點，假設不存在重複元素 * @param <T> */
class TreeNode<T extends Comparable<T>> {
    T value;
    TreeNode<T> parent;
    TreeNode<T> left, right;

    public TreeNode(T value, TreeNode<T> parent) {
        this.value = value;
        this.parent = parent;
    }

    /** * 獲取根節點 */
    TreeNode<T> root() {
        TreeNode<T> cur = this;
        while (cur.parent != null) {
            cur = cur.parent;
        }
        return cur;
    }

    /** * 中序遍歷 */
    void inOrderTraverse() {
        if(this.left != null) this.left.inOrderTraverse();
        System.out.println(this.value);
        if(this.right != null) this.right.inOrderTraverse();
    }

    /** * 經典的二叉樹插入元素的方法 */
    TreeNode<T> insert(T value) {
        // 先找根元素
        TreeNode<T> cur = root();

        TreeNode<T> p;
        int dir;

        // 尋找元素應該插入的位置
        do {
            p = cur;
            if ((dir=value.compareTo(p.value)) < 0) {
                cur = cur.left;
            } else {
                cur = cur.right;
            }
        } while (cur != null);

        // 把元素放到找到的位置
        if (dir < 0) {
            p.left = new TreeNode<>(value, p);
            return p.left;
        } else {
            p.right = new TreeNode<>(value, p);
            return p.right;
        }
    }
}
複製代碼

TreeMap的遍歷

從上面二叉樹的遍歷咱們很明顯地看到，它是經過遞歸的方式實現的，可是遞歸會佔用額外的空間，直接到線程棧整個釋放掉纔會把方法中申請的變量銷燬掉，因此當元素特別多的時候是一件很危險的事。ui

（上面的例子中，沒有申請額外的空間，若是有聲明變量，則能夠理解爲直到方法完成纔會銷燬變量）this

那麼，有沒有什麼方法不用遞歸呢？spa

讓咱們來看看java中的實現：

@Override
public void forEach(BiConsumer<? super K, ? super V> action) {
    Objects.requireNonNull(action);
    // 遍歷前的修改次數
    int expectedModCount = modCount;
    // 執行遍歷，先獲取第一個元素的位置，再循環遍歷後繼節點
    for (Entry<K, V> e = getFirstEntry(); e != null; e = successor(e)) {
        // 執行動做
        action.accept(e.key, e.value);

        // 若是發現修改次數變了，則拋出異常
        if (expectedModCount != modCount) {
            throw new ConcurrentModificationException();
        }
    }
}
複製代碼

是否是很簡單？！

（1）尋找第一個節點；

從根節點開始找最左邊的節點，即最小的元素。

final Entry<K,V> getFirstEntry() {
        Entry<K,V> p = root;
        // 從根節點開始找最左邊的節點，即最小的元素
        if (p != null)
            while (p.left != null)
                p = p.left;
        return p;
    }
複製代碼

（2）循環遍歷後繼節點；

尋找後繼節點這個方法咱們在刪除元素的時候也用到過，當時的場景是有右子樹，則從其右子樹中尋找最小的節點。

static <K,V> TreeMap.Entry<K,V> successor(Entry<K,V> t) {
    if (t == null)
        // 若是當前節點爲空，返回空
        return null;
    else if (t.right != null) {
        // 若是當前節點有右子樹，取右子樹中最小的節點
        Entry<K,V> p = t.right;
        while (p.left != null)
            p = p.left;
        return p;
    } else {
        // 若是當前節點沒有右子樹
        // 若是當前節點是父節點的左子節點，直接返回父節點
        // 若是當前節點是父節點的右子節點，一直往上找，直到找到一個祖先節點是其父節點的左子節點爲止，返回這個祖先節點的父節點
        Entry<K,V> p = t.parent;
        Entry<K,V> ch = t;
        while (p != null && ch == p.right) {
            ch = p;
            p = p.parent;
        }
        return p;
    }
}
複製代碼