二叉樹的非遞歸遍歷自己理解版本

**

二叉樹的非遞歸遍歷

**
中序:
中序遍歷是先左孩子 後跟 最後右孩子的順序。所以應該先一個while循環把所有的根及其左孩子依次入棧,然後一個while循環 將棧頂元素取出,遍歷,如果有右孩子,同樣需要將其及其左孩子入棧。代碼如下

    TreeNode tem = root;
    LinkedList<TreeNode> stack = new LinkedList<>();
    while(tem != null){
        stack.push(tem);
        tem = tem.left;
    }
    while(!stack.isEmpty()){
        TreeNode node = stack.pop();
        //假裝遍歷一下
        System.out.print(node.val);
        node = node.right;
        while (node!=null){
            stack.push(node);
            node = node.left;
        }
    }

前序遍歷:
前序遍歷是先跟 然後左右孩子。所以直接將非null根結點入棧,然後一個while循環 將棧頂元素取出 遍歷,如果有右孩子入隊,有左孩子 入隊,代碼如下
if(root!=null){
LinkedList stack = new LinkedList<>();
stack.push(root);
while(!stack.isEmpty()){
TreeNode node = stack.pop();
//假裝遍歷
if(node.right != null){
stack.push(node.right);
}
if(node.left != null){
stack.push(node.left);
}

        }
    }

後序遍歷:
後序遍歷其實是相對較難的。但是也比較好理解
後序就是先左右孩子然後跟,所以我們任然需要將跟及其左孩子依次入棧,不同的是我們不能直接遍歷,而是peek不是pop拿到到棧頂元素,如果該元素有右孩子並且未被訪問過,那麼我們就將右孩子入棧,並將其左孩子依次入棧。否則 訪問該節點並更新剛剛訪問過的結點到last 變量中(該變量的作用是判斷是否被訪問過的根據,如果一個元素
右孩子爲剛剛訪問過的,那麼就不需要將右孩子及右孩子的左孩子依次入棧了)。代碼如下

List res = new ArrayList<>();
TreeNode last = null;
if(root!=null){

        LinkedList<TreeNode> stack = new LinkedList<>();
        while(root!=null){
            stack.push(root);
            root = root.left;
        }
        while (!stack.isEmpty()){
            TreeNode node = stack.peek();

            if(node.right != null && last != node.right){

                TreeNode tem = node.right;
                stack.push(tem);
                while(tem.left!=null){
                    stack.push(tem.left);
                    tem = tem.left;
                }

            }else{
                TreeNode node1 = stack.pop();
                last = node1;
                res.add(node1.val);

            }

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