@graphty/algorithms/dist/src/algorithms/mst/kruskal.js

Graph algorithms library for browser environments implemented in TypeScript

import { UnionFind } from "../../data-structures/union-find.js";
export function kruskalMST(graph) {
    if (graph.isDirected) {
        throw new Error("Kruskal's algorithm requires an undirected graph");
    }
    const edges = [];
    const visitedEdges = new Set();
    for (const edge of Array.from(graph.edges())) {
        const edgeKey = edge.source < edge.target ?
            `${String(edge.source)}-${String(edge.target)}` :
            `${String(edge.target)}-${String(edge.source)}`;
        if (!visitedEdges.has(edgeKey)) {
            visitedEdges.add(edgeKey);
            edges.push(edge);
        }
    }
    edges.sort((a, b) => (a.weight ?? 0) - (b.weight ?? 0));
    const nodes = Array.from(graph.nodes()).map((n) => n.id);
    const uf = new UnionFind(nodes);
    const mstEdges = [];
    let totalWeight = 0;
    for (const edge of edges) {
        if (!uf.connected(edge.source, edge.target)) {
            uf.union(edge.source, edge.target);
            mstEdges.push(edge);
            totalWeight += edge.weight ?? 0;
            if (mstEdges.length === graph.nodeCount - 1) {
                break;
            }
        }
    }
    if (mstEdges.length !== graph.nodeCount - 1) {
        throw new Error("Graph is not connected");
    }
    return {
        edges: mstEdges,
        totalWeight,
    };
}
export function minimumSpanningTree(graph) {
    return kruskalMST(graph);
}
//# sourceMappingURL=kruskal.js.map