Core graph data structures and basic operations for the yog library.
This module defines the fundamental Graph type and provides all basic operations
for creating and manipulating graphs. The graph uses an adjacency list representation
with dual indexing (both outgoing and incoming edges) for efficient traversal in both
directions.
Graph Types
- Directed Graph: Edges have a direction (one-way relationships)
- Undirected Graph: Edges are bidirectional (mutual relationships)
Type Parameters
node_data: The type of data stored at each node (e.g.,String,City,Task)edge_data: The type of data stored on edges, typically weights (e.g.,Int,Float)
Quick Start
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge!(1, 2, 10)
iex> Yog.Model.successors(graph, 1)
[{2, 10}]
iex> Yog.Model.node(graph, 1)
"A"Design Notes
The dual-map representation enables O(1) edge existence checks and O(1) transpose operations, at the cost of increased memory usage and slightly more complex edge updates.
Summary
Types
A simple graph data structure that can be directed or undirected.
The type of graph: :directed or :undirected.
Unique identifier for a node in the graph.
Functions
Adds an edge to the graph.
Adds an edge to the graph with the given weight.
Adds an edge to the graph, raising on error.
Same as add_edge/4 but raises on error.
Ensures both endpoint nodes exist, then adds an edge.
Ensures both endpoint nodes exist, then adds an edge.
Ensures both endpoint nodes exist using a callback, then adds an edge.
Adds an edge, but if an edge already exists between src and dst,
it combines the new weight with the existing one using with_combine.
Same as add_edge_with_combine/5 but raises on error.
Adds multiple edges to the graph in a single operation.
Adds multiple edges to the graph, raising on error.
Adds a node to the graph with the given ID and data. If a node with this ID already exists, its data will be replaced.
Adds multiple nodes to the graph from an iterable.
Adds multiple simple edges (weight = 1) to the graph.
Adds an unweighted edge to the graph.
Adds multiple unweighted edges (weight = nil) to the graph.
Returns all edges in the graph as triplets {from, to, weight}.
Returns all node IDs in the graph. This includes all nodes, even isolated nodes with no edges.
Returns the total degree of a node.
Returns the number of edges in the graph.
Gets the weight/data of an edge between two nodes.
Returns nil if the edge doesn't exist.
Gets the weight/data of an edge between two nodes, raising if not found.
Checks if the graph contains an edge between src and dst.
Checks if the graph contains a node with the given ID.
Returns the in-degree of a node (number of incoming edges).
Returns all neighbor node IDs (without weights).
Gets all nodes connected to the given node, regardless of direction. Useful for algorithms like finding "connected components".
Creates a new empty graph of the specified type.
Gets the data associated with a node.
Returns the number of nodes in the graph.
Equivalent to order/1.
Returns all nodes in the graph as a map.
Returns the number of nodes in the graph (graph order).
Returns the out-degree of a node (number of outgoing edges).
Returns all predecessor node IDs (without weights).
Gets nodes you came FROM to reach the given node (predecessors). Returns a list of tuples containing the source node ID and edge data.
Removes a directed edge from src to dst.
Removes a node and all its connected edges (incoming and outgoing).
Returns all successor node IDs (without weights).
Gets nodes you can travel TO from the given node (successors). Returns a list of tuples containing the destination node ID and edge data.
Gets the type of the graph (:directed or :undirected).
Types
@type graph() :: Yog.Graph.t()
A simple graph data structure that can be directed or undirected.
This is an alias for Yog.Graph.t().
@type graph_type() :: :directed | :undirected
The type of graph: :directed or :undirected.
@type node_id() :: term()
Unique identifier for a node in the graph.
Functions
Adds an edge to the graph.
For directed graphs, adds a single edge from from to to.
For undirected graphs, adds edges in both directions.
Returns {:ok, graph} or {:error, reason}.
Example
iex> {:ok, graph} =
...> Yog.directed()
...> |> Yog.add_node(1, "A")
...> |> Yog.add_node(2, "B")
...> |> Yog.add_edge(from: 1, to: 2, with: 10)
iex> Yog.successors(graph, 1)
[{2, 10}]With pattern matching for chaining
iex> graph = Yog.directed() |> Yog.add_node(1, "A") |> Yog.add_node(2, "B")
iex> {:ok, graph} = Yog.add_edge(graph, from: 1, to: 2, with: 10)
iex> {:ok, graph} = Yog.add_edge(graph, from: 2, to: 1, with: 5)
iex> Yog.successors(graph, 2)
[{1, 5}]
Adds an edge to the graph with the given weight.
For directed graphs, adds a single edge from src to dst.
For undirected graphs, adds edges in both directions.
Returns {:error, reason} if either endpoint node doesn't exist in graph.nodes.
Use add_edge_ensure/5 to auto-create missing nodes with a default value,
or add_node/3 to explicitly add nodes before adding edges.
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge(1, 2, 10)
iex> Yog.Model.successors(graph, 1)
[{2, 10}]
iex> # Error when nodes don't exist
iex> Yog.Model.new(:directed) |> Yog.Model.add_edge(1, 2, 10)
{:error, "Nodes 1 and 2 do not exist"}
Adds an edge to the graph, raising on error.
Example
iex> graph = Yog.directed() |> Yog.add_node(1, "A") |> Yog.add_node(2, "B")
iex> graph = Yog.add_edge!(graph, from: 1, to: 2, with: 10)
iex> Yog.successors(graph, 1)
[{2, 10}]
Same as add_edge/4 but raises on error.
Example
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge!(1, 2, 10)
iex> Yog.Model.successors(graph, 1)
[{2, 10}]
Ensures both endpoint nodes exist, then adds an edge.
If from or to is not already in the graph, it is created with
the supplied default node data. Existing nodes are left unchanged.
Always succeeds and returns a Graph (never fails).
Use this when you want to build graphs quickly without pre-creating nodes.
Example
iex> graph =
...> Yog.directed()
...> |> Yog.add_edge_ensure(from: 1, to: 2, with: 10, default: "anon")
iex> # Nodes 1 and 2 are auto-created with data "anon"
iex> Yog.successors(graph, 1)
[{2, 10}]
Ensures both endpoint nodes exist, then adds an edge.
If src or dst is not already in the graph, it is created with
the supplied default node data before the edge is added. Nodes
that already exist are left unchanged. If no default is submitted nil
is used as the default node data.
Always succeeds and returns a Graph (never fails).
Example
iex> # Nodes 1 and 2 are created automatically with data "anon"
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_edge_ensure(1, 2, 10, "anon")
iex> Yog.Model.node(graph, 1)
"anon"
iex> # Existing nodes keep their data
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_edge_ensure(1, 2, 5, "B")
iex> {Yog.Model.node(graph, 1), Yog.Model.node(graph, 2)}
{"A", "B"}
Ensures both endpoint nodes exist using a callback, then adds an edge.
If src or dst is not already in the graph, it is created by
calling the by function with the node ID to generate the node data.
Nodes that already exist are left unchanged.
Always succeeds and returns a Graph (never fails).
Example
iex> # Nodes 1 and 2 are created automatically with value that's the same as NodeId
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_edge_with(1, 2, 10, fn x -> x end)
iex> Yog.Model.node(graph, 1)
1
iex> # Existing nodes keep their data; only missing ones get generated data
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "1")
...> |> Yog.Model.add_edge_with(1, 2, 5, fn n -> to_string(n) <> ":new" end)
iex> # Node 1 is still "1", node 2 is "2:new"
iex> {Yog.Model.node(graph, 1), Yog.Model.node(graph, 2)}
{"1", "2:new"}
@spec add_edge_with_combine(graph(), node_id(), node_id(), term(), (term(), term() -> term())) :: {:ok, graph()} | {:error, String.t()}
Adds an edge, but if an edge already exists between src and dst,
it combines the new weight with the existing one using with_combine.
The combine function receives (existing_weight, new_weight) and should
return the combined weight.
Returns {:error, reason} if either endpoint node doesn't exist in graph.nodes.
Time Complexity: O(1)
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge(1, 2, 10)
iex> {:ok, graph} = Yog.Model.add_edge_with_combine(graph, 1, 2, 5, &Kernel.+/2)
iex> # Edge 1->2 now has weight 15 (10 + 5)
iex> Yog.Model.successors(graph, 1)
[{2, 15}]Use Cases
- Edge contraction in graph algorithms (Stoer-Wagner min-cut)
- Multi-graph support (adding parallel edges with combined weights)
- Incremental graph building (accumulating weights from multiple sources)
@spec add_edge_with_combine!(graph(), node_id(), node_id(), term(), (term(), term() -> term())) :: graph()
Same as add_edge_with_combine/5 but raises on error.
Example
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge!(1, 2, 10)
...> |> Yog.Model.add_edge_with_combine!(1, 2, 5, &Kernel.+/2)
iex> Yog.Model.successors(graph, 1)
[{2, 15}]
Adds multiple edges to the graph in a single operation.
Fails fast on the first edge that references non-existent nodes.
Returns {:error, reason} if any endpoint node doesn't exist.
This is more ergonomic than chaining multiple add_edge calls
as it only requires unwrapping a single Result.
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_node(3, "C")
...> |> Yog.Model.add_edges([{1, 2, 10}, {2, 3, 5}, {1, 3, 15}])
iex> length(Yog.Model.successors(graph, 1))
2
Adds multiple edges to the graph, raising on error.
Example
iex> graph =
...> Yog.directed()
...> |> Yog.add_node(1, "A")
...> |> Yog.add_node(2, "B")
...> |> Yog.add_node(3, "C")
...> |> Yog.add_edges!([{1, 2, 10}, {2, 3, 5}])
iex> length(Yog.successors(graph, 1))
1
Adds a node to the graph with the given ID and data. If a node with this ID already exists, its data will be replaced.
Example
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "Node A")
...> |> Yog.Model.add_node(2, "Node B")
iex> Yog.Model.order(graph)
2
@spec add_nodes_from(graph(), Enumerable.t()) :: graph()
Adds multiple nodes to the graph from an iterable.
Accepts:
- A list of node IDs:
[1, 2, 3] - A list of
{id, data}tuples:[{1, "A"}, {2, "B"}] - A map:
%{1 => "A", 2 => "B"} - Another
Yog.Graph: copies all nodes (but not edges)
Existing nodes with the same ID will have their data replaced.
Example
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_nodes_from([1, {2, "B"}])
iex> Yog.Model.order(graph)
2
iex> Yog.Model.node(graph, 2)
"B"
Adds multiple simple edges (weight = 1) to the graph.
Fails fast on the first edge that references non-existent nodes. Convenient for unweighted graphs where all edges have weight 1.
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_node(3, "C")
...> |> Yog.Model.add_simple_edges([{1, 2}, {2, 3}, {1, 3}])
iex> Yog.Model.successors(graph, 1) |> Enum.sort()
[{2, 1}, {3, 1}]
Adds an unweighted edge to the graph.
This is a convenience function for graphs where edges have no meaningful weight.
Uses nil as the edge data type.
Returns {:ok, graph} or {:error, reason} if either endpoint node doesn't exist.
Example
iex> {:ok, graph} =
...> Yog.directed()
...> |> Yog.add_node(1, "A")
...> |> Yog.add_node(2, "B")
...> |> Yog.add_unweighted_edge(from: 1, to: 2)
iex> Yog.successors(graph, 1)
[{2, nil}]
@spec add_unweighted_edges(graph(), [{node_id(), node_id()}]) :: {:ok, graph()} | {:error, String.t()}
Adds multiple unweighted edges (weight = nil) to the graph.
Fails fast on the first edge that references non-existent nodes. Convenient for graphs where edges carry no weight information.
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_node(3, "C")
...> |> Yog.Model.add_unweighted_edges([{1, 2}, {2, 3}, {1, 3}])
iex> Yog.Model.successors(graph, 1) |> Enum.sort()
[{2, nil}, {3, nil}]
Returns all edges in the graph as triplets {from, to, weight}.
For directed graphs, returns all edges.
For undirected graphs, returns each edge only once (where from <= to).
This is particularly useful for graph export formats.
Examples
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, nil)
...> |> Yog.Model.add_node(2, nil)
...> |> Yog.Model.add_edge!(1, 2, 5)
iex> Yog.Model.all_edges(graph)
[{1, 2, 5}]
iex> graph =
...> Yog.Model.new(:undirected)
...> |> Yog.Model.add_node(1, nil)
...> |> Yog.Model.add_node(2, nil)
...> |> Yog.Model.add_edge!(1, 2, 5)
iex> edges = Yog.Model.all_edges(graph)
iex> length(edges)
1
Returns all node IDs in the graph. This includes all nodes, even isolated nodes with no edges.
Example
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
iex> Yog.Model.all_nodes(graph) |> Enum.sort()
[1, 2]
@spec degree(graph(), node_id()) :: non_neg_integer()
Returns the total degree of a node.
For directed graphs, this is the sum of in-degree and out-degree. For undirected graphs, this counts each edge once (self-loops count as 2).
Time Complexity: O(1) for undirected, O(1) for directed
Example
iex> {:ok, graph} =
...> Yog.Model.new(:undirected)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge(1, 2, 10)
iex> Yog.Model.degree(graph, 1)
1
Returns the number of edges in the graph.
For undirected graphs, each edge is counted once (the pair {u, v}). For directed graphs, each directed edge (u -> v) is counted once.
Time Complexity: O(V)
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge(1, 2, 10)
iex> Yog.Model.edge_count(graph)
1
Gets the weight/data of an edge between two nodes.
Returns nil if the edge doesn't exist.
Example
iex> graph = Yog.directed() |> Yog.add_edge_ensure(1, 2, 10, nil)
iex> Yog.Model.edge_data(graph, 1, 2)
10
Gets the weight/data of an edge between two nodes, raising if not found.
Example
iex> graph = Yog.directed() |> Yog.add_edge_ensure(1, 2, 10, nil)
iex> Yog.Model.edge_data!(graph, 1, 2)
10
Checks if the graph contains an edge between src and dst.
Returns true if an edge exists, false otherwise.
Examples
iex> graph = Yog.from_edges(:directed, [{1, 2, 10}, {1, 3, 20}, {3, 4, 2}])
iex> Yog.Model.has_edge?(graph, 1, 2)
true
iex> Yog.Model.has_edge?(graph, 2, 1)
false
iex> graph = Yog.from_edges(:undirected, [{1, 2, 10}, {1, 3, 20}])
iex> Yog.Model.has_edge?(graph, 2, 1)
true
iex> Yog.Model.has_edge?(graph, 2, 4)
falseTime Complexity: O(1)
Checks if the graph contains a node with the given ID.
Example
iex> graph = Yog.undirected() |> Yog.add_node(1, nil)
iex> Yog.Model.has_node?(graph, 1)
true
iex> Yog.Model.has_node?(graph, 2)
falseTime Complexity: O(1)
@spec in_degree(graph(), node_id()) :: non_neg_integer()
Returns the in-degree of a node (number of incoming edges).
For undirected graphs, this returns the total degree (same as out_degree/2).
Time Complexity: O(1)
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge(1, 2, 10)
iex> Yog.Model.in_degree(graph, 2)
1
iex> Yog.Model.in_degree(graph, 1)
0
Returns all neighbor node IDs (without weights).
Example
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge_ensure(1, 2, 10)
...> |> Yog.Model.add_edge_ensure(1, 3, 20)
iex> Yog.Model.neighbor_ids(graph, 1) |> Enum.sort()
[2, 3]
Gets all nodes connected to the given node, regardless of direction. Useful for algorithms like finding "connected components".
For undirected graphs, this is equivalent to successors. For directed graphs, this combines successors and predecessors without duplicates.
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_node(3, "C")
...> |> Yog.Model.add_edges([{1, 2, 10}, {3, 1, 20}])
iex> Yog.Model.neighbors(graph, 1) |> Enum.sort()
[{2, 10}, {3, 20}]
@spec new(graph_type()) :: graph()
Creates a new empty graph of the specified type.
Example
iex> graph = Yog.Model.new(:directed)
iex> Yog.Model.order(graph)
0
Gets the data associated with a node.
Returns nil if the node doesn't exist.
Example
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
iex> Yog.Model.node(graph, 1)
"A"
Returns the number of nodes in the graph.
Equivalent to order/1.
Time Complexity: O(1)
Example
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
iex> Yog.Model.node_count(graph)
2
Returns all nodes in the graph as a map.
Example
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
iex> nodes = Yog.Model.nodes(graph)
iex> nodes[1]
"A"
Returns the number of nodes in the graph (graph order).
Time Complexity: O(1)
Example
iex> graph =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
iex> Yog.Model.order(graph)
2
@spec out_degree(graph(), node_id()) :: non_neg_integer()
Returns the out-degree of a node (number of outgoing edges).
For undirected graphs, this returns the total degree (same as in_degree/2).
Time Complexity: O(1)
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge(1, 2, 10)
iex> Yog.Model.out_degree(graph, 1)
1
iex> Yog.Model.out_degree(graph, 2)
0
Returns all predecessor node IDs (without weights).
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge(1, 2, 10)
iex> Yog.Model.predecessor_ids(graph, 2)
[1]
Gets nodes you came FROM to reach the given node (predecessors). Returns a list of tuples containing the source node ID and edge data.
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge(1, 2, 10)
iex> Yog.Model.predecessors(graph, 2)
[{1, 10}]
Removes a directed edge from src to dst.
For directed graphs, this removes the single directed edge from src to dst.
For undirected graphs, this removes the edges in both directions
(from src to dst and from dst to src).
Time Complexity: O(1)
Example
iex> # Directed graph - removes single directed edge
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge(1, 2, 10)
iex> graph = Yog.Model.remove_edge(graph, 1, 2)
iex> # Edge 1->2 is removed
iex> Yog.Model.successors(graph, 1)
[]
iex> # Undirected graph - removes both directions
iex> {:ok, graph} =
...> Yog.Model.new(:undirected)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge(1, 2, 10)
iex> graph = Yog.Model.remove_edge(graph, 1, 2)
iex> # Edge between 1 and 2 is fully removed
iex> Yog.Model.successors(graph, 1)
[]
Removes a node and all its connected edges (incoming and outgoing).
Time Complexity: O(deg(v)) - proportional to the number of edges connected to the node, not the whole graph.
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_node(3, "C")
...> |> Yog.Model.add_edges([{1, 2, 10}, {2, 3, 20}])
iex> graph = Yog.Model.remove_node(graph, 2)
iex> # Node 2 is removed, along with edges 1->2 and 2->3
iex> Yog.Model.order(graph)
2
Returns all successor node IDs (without weights).
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge(1, 2, 10)
iex> Yog.Model.successor_ids(graph, 1)
[2]
Gets nodes you can travel TO from the given node (successors). Returns a list of tuples containing the destination node ID and edge data.
Example
iex> {:ok, graph} =
...> Yog.Model.new(:directed)
...> |> Yog.Model.add_node(1, "A")
...> |> Yog.Model.add_node(2, "B")
...> |> Yog.Model.add_edge(1, 2, 10)
iex> Yog.Model.successors(graph, 1)
[{2, 10}]
@spec type(graph()) :: graph_type()
Gets the type of the graph (:directed or :undirected).
Example
iex> graph = Yog.Model.new(:directed)
iex> Yog.Model.type(graph)
:directed