joshkaposh-graph/src/visit/visitor.ts

A collection of general purpose graph data structures and traversal algorithms

import { FixedBitSet } from "fixed-bit-set";
import { VisitMap } from "./traversal";

export class VisitorSet<N> implements VisitMap<N> {
    #set: Set<N>;
    constructor() {
        this.#set = new Set()
    }

    len(): number {
        return this.#set.size;
    }

    visit(x: N): boolean {
        const is_first_visit = !this.#set.has(x);
        this.#set.add(x)
        return is_first_visit;
    }

    is_visited(x: N): boolean {
        return this.#set.has(x);
    }

    clear() {
        this.#set.clear();
    }
}

export class VisitorFbs<N extends number> implements VisitMap<N> {
    #fbs: FixedBitSet;
    constructor(node_count: number) {
        this.#fbs = FixedBitSet.with_capacity(node_count)
    }

    len(): number {
        return this.#fbs.len()
    }

    clear() {
        this.#fbs.clear();
    }

    grow(bits: number) {
        this.#fbs.grow(bits)
    }

    visit(x: N): boolean {
        return !this.#fbs.put(x);
    }

    is_visited(x: N): boolean {
        return this.#fbs.contains(x);
    }
}