web/ranked_tree.js

class Queue {
    constructor() {
        this.q = [];
    }

    enqueue(x) {
        this.q.push(x);
    }

    dequeue() {
        return this.q.shift();
    }

    isEmpty() {
        return this.q.length == 0;
    }
}

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

    _getUnwrappedGraph() {
        let result = [];

        let q = new Queue();
        q.enqueue(this.root);

        let edges = [];
        while (!q.isEmpty()) {
            let node = q.dequeue();
            if (!q) {
                continue;
            }

            result.push(
                `\t"${node.toString()}" [label="${node.value}, ${node.rank}"];`
            );

            [node.left, node.right]
                .filter((child) => child)
                .forEach((child) => {
                    edges.push([node, child]);
                    q.enqueue(child);
                });
        }

        result.push("");

        edges.forEach((vertices) => {
            let [u, v] = vertices;
            result.push(
                `\t"${u.toString()}" -> "${v.toString()}" [label="${nodeDifference(
                    v
                )}"]`
            );
        });

        return result;
    }

    toDot() {
        return ["digraph {", ...this._getUnwrappedGraph(), "}"];
    }

    toString() {
        return this.toDot().join("\n");
    }

    record() {
        if (!this.recorder) {
            return;
        }
        this.recorder.record(this);
    }

    rank() {
        return nodeRank(this.root);
    }

    isCorrect() {
        return this.isCorrectNode(this.root);
    }

    search(value, node) {
        return nodeSearch(value, node || this.root);
    }

    insert(value) {
        let insertedNode = new Node(value);

        if (!this.root) {
            this.root = insertedNode;
            this.record();
            return;
        }

        let parent = findParentNode(value, this.root);
        if (!parent) {
            return;
        }

        insertedNode.parent = parent;

        if (value < parent.value) {
            parent.left = insertedNode;
        } else {
            parent.right = insertedNode;
        }
        this.record();

        this.insertRebalance(insertedNode);
    }

    transplant(u, v) {
        if (!u.parent) {
            this.root = v;
        } else if (u.parent.left === u) {
            u.parent.left = v;
        } else {
            u.parent.right = v;
        }

        if (v) {
            v.rank = u.rank;
            v.parent = u.parent;
        }
    }

    deleteNode(node) {
        if (!node) {
            return null;
        }

        let [y, parent] = [null, node.parent];

        if (!node.left) {
            y = node.right;
            this.transplant(node, node.right);
        } else if (!node.right) {
            y = node.left;
            this.transplant(node, node.left);
        } else {
            let successor = nodeMinimum(node.right);
            parent = successor.parent != node ? successor.parent : successor;

            if (successor.parent != node) {
                parent = successor.right ? successor.right : successor.parent;
                this.transplant(successor, successor.right);
                successor.right = node.right;
                successor.right.parent = successor;
            }

            this.transplant(node, successor);
            successor.left = node.left;
            node.left.parent = successor;
        }

        this.record();

        return [y, parent];
    }

    delete(value) {
        let node = this.root;
        while (node && node.value != value) {
            node = value < node.value ? node.left : node.right;
        }

        let toRebalance = this.deleteNode(node);
        if (toRebalance) {
            let [y, parent] = toRebalance;
            this.deleteRebalance(y, parent);
        }
    }

    /* abstract methods */
    isCorrectNode(node) {
        throw "not implemented!";
    }

    insertRebalance(node) {
        throw "not implemented!";
    }

    deleteRebalance(node, parent) {
        throw "not implemented!";
    }
}