52 # 二叉树的前中后遍历

二叉树的遍历

线性数据结构遍历比较简单,可以采用正序遍历、逆序遍历。

遍历树的目的一般是修改树,比如修改树的节点,采用访问者模式

前序遍历

前序遍历(preorder traversal):先访问根节点,前序遍历左子树,前序遍历右子树;遍历 dom 树可以使用

在这里插入图片描述

perorderTraversal(visitor) {
    
    
    if (visitor == null) return;
    const traversal = (node) => {
    
    
        if (node === null) return;
        visitor.visit(node);
        traversal(node.left);
        traversal(node.right);
    };
    traversal(this.root);
}

在这里插入图片描述

中序遍历

中序遍历(inorder traversal):中序遍历左子树,根节点,中序遍历右子树;可以顺序打印数据

在这里插入图片描述

inorderTraversal(visitor) {
    
    
    if (visitor == null) return;
    const traversal = (node) => {
    
    
        if (node === null) return;
        traversal(node.left);
        visitor.visit(node);
        traversal(node.right);
    };
    traversal(this.root);
}

在这里插入图片描述

后序遍历

后序遍历(postorder traversal):后序遍历左子树,后续遍历右子树,根节点;遍历子节点可使用后序遍历

在这里插入图片描述

postorderTraversal(visitor) {
    
    
    if (visitor == null) return;
    const traversal = (node) => {
    
    
        if (node === null) return;
        traversal(node.left);
        traversal(node.right);
        visitor.visit(node);
    };
    traversal(this.root);
}

在这里插入图片描述

层序遍历

层序遍历(level order traversal):从上到下,从左到右依次访问每一个节点

在这里插入图片描述

完整代码

实现如下:

// 节点
class Node {
    
    
    constructor(element, parent) {
    
    
        this.element = element; // 存的数据
        this.parent = parent; // 父节点
        this.left = null; // 左子树
        this.right = null; // 右子树
    }
}

class BST {
    
    
    constructor(compare) {
    
    
        this.root = null;
        this.size = 0; // 节点个数
        this.compare = compare || this.compare;
    }
    compare(e1, e2) {
    
    
        return e1 - e2;
    }
    // 添加节点
    add(element) {
    
    
        // 如果根元素不存在
        if (this.root === null) {
    
    
            this.root = new Node(element, null);
            this.size++;
            return;
        } else {
    
    
            // 如果根元素存在,那么增加的就不是根节点,需要找到 parent
            let currentNode = this.root;
            // 当前比较的结果
            let compare = 0;
            // 先找到需要对比的 parent(当前父节点)
            let parent = null;
            while (currentNode) {
    
    
                parent = currentNode;
                compare = this.compare(element, currentNode.element);
                // 如果大于 0 找右树,小于 0 找左树
                if (compare > 0) {
    
    
                    currentNode = currentNode.right;
                } else if (compare < 0) {
    
    
                    currentNode = currentNode.left;
                } else {
    
    
                    // 如果比较后结果一样,由自己决定是否需要覆盖
                    currentNode.element = element; // 覆盖
                    return;
                }
            }
            // 找到了 parent,生成新节点
            let newNode = new Node(element, parent);
            if (compare > 0) {
    
    
                parent.right = newNode;
            } else {
    
    
                parent.left = newNode;
            }
        }
    }
    // 先序遍历
    perorderTraversal(visitor) {
    
    
        if (visitor == null) return;
        const traversal = (node) => {
    
    
            if (node === null) return;
            visitor.visit(node);
            traversal(node.left);
            traversal(node.right);
        };
        traversal(this.root);
    }
    // 中序遍历
    inorderTraversal(visitor) {
    
    
        if (visitor == null) return;
        const traversal = (node) => {
    
    
            if (node === null) return;
            traversal(node.left);
            visitor.visit(node);
            traversal(node.right);
        };
        traversal(this.root);
    }
    // 后序遍历
    postorderTraversal(visitor) {
    
    
        if (visitor == null) return;
        const traversal = (node) => {
    
    
            if (node === null) return;
            traversal(node.left);
            traversal(node.right);
            visitor.visit(node);
        };
        traversal(this.root);
    }
}

let bst = new BST();

let arr = [10, 8, 19, 6, 15, 22, 20];

arr.forEach((element) => {
    
    
    bst.add(element);
});

// 遍历树的目的一般是修改树,比如修改树的节点,采用访问者模式
console.log(
    bst.perorderTraversal({
    
    
        visit(node) {
    
    
            console.log("visitor.visit---->", node.element);
        },
    })
);
console.log(
    bst.inorderTraversal({
    
    
        visit(node) {
    
    
            console.log("visitor.visit---->", node.element);
        },
    })
);
console.log(
    bst.postorderTraversal({
    
    
        visit(node) {
    
    
            console.log("visitor.visit---->", node.element);
        },
    })
);

猜你喜欢

转载自blog.csdn.net/kaimo313/article/details/131661922
52