Tree

Directions

Create a Node class.
- The constructor should accept an argument that gets assigned to the data property and initialize an empty array for storing children.
- The Node class should have methods add and remove.
Create a Tree class.
- The constructor should initialize a root property to null.
Implement traverseBF (breadth first) and traverseDepthFirst (depth first) on the Tree class.
- Each method should accept a function that gets called with each element in the tree.

Solutions

JS

Node class

class Node {
    constructor(data) {
        this.data = data;
        this.children = [];
    }

    add(data) {
        this.children.push(new Node(data));
    }

    remove(data) {
        this.children = this.children.filter((node) => {
            return node.data !== data;
        })
    }
}

Tree class

class Tree {
    constructor() {
        this.root = null;
    }

    /**
     * @method traverseBF
     * @description Executes a breadth-first traversal of the tree.
     */
    traverseBF(fn) {
        const arr = [this.root];

        while (arr.length > 0) {
            const node = arr.shift();

            arr.push(...node.children);
            fn(node);
        }
    }

    /**
     * @method traverseDF
     * @description Executes a depth-first traversal of the tree.
     */
    traverseDF(fn) {
        const arr = [this.root];

        while (arr.length > 0) {
            const node = arr.shift();

            arr.unshift(...node.children);
            fn(node);
        }
    }
}

Made with