mirror of
https://github.com/TheAlgorithms/JavaScript.git
synced 2025-07-05 00:01:37 +08:00
Added Connected Components using DFS (#223)
This commit is contained in:
56
Graphs/ConnectedComponents.js
Normal file
56
Graphs/ConnectedComponents.js
Normal file
@ -0,0 +1,56 @@
|
||||
class GraphUnweightedUndirected {
|
||||
// Unweighted Undirected Graph class
|
||||
constructor () {
|
||||
this.connections = {}
|
||||
}
|
||||
|
||||
addNode (node) {
|
||||
// Function to add a node to the graph (connection represented by set)
|
||||
this.connections[node] = new Set()
|
||||
}
|
||||
|
||||
addEdge (node1, node2) {
|
||||
// Function to add an edge (adds the node too if they are not present in the graph)
|
||||
if (!(node1 in this.connections)) { this.addNode(node1) }
|
||||
if (!(node2 in this.connections)) { this.addNode(node2) }
|
||||
this.connections[node1].add(node2)
|
||||
this.connections[node2].add(node1)
|
||||
}
|
||||
|
||||
DFSComponent (components, node, visited) {
|
||||
// Helper function to populate the visited set with the nodes in each component
|
||||
|
||||
// adding the first visited node in the component to the array
|
||||
components.push(node)
|
||||
const stack = [node]
|
||||
// populating the visited set using DFS (Iterative)
|
||||
while (stack.length > 0) {
|
||||
const curr = stack.pop()
|
||||
visited.add(curr.toString())
|
||||
for (const neighbour of this.connections[curr].keys()) {
|
||||
if (!visited.has(neighbour.toString())) { stack.push(neighbour) }
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
connectedComponents () {
|
||||
// Function to generate the Connected Components
|
||||
// Result is an array containing 1 node from each component
|
||||
const visited = new Set()
|
||||
const components = []
|
||||
for (const node of Object.keys(this.connections)) {
|
||||
if (!visited.has(node.toString())) { this.DFSComponent(components, node, visited) }
|
||||
}
|
||||
return components
|
||||
}
|
||||
}
|
||||
|
||||
function main () {
|
||||
const graph = new GraphUnweightedUndirected()
|
||||
graph.addEdge(1, 2) // Component 1
|
||||
graph.addEdge(3, 4) // Component 2
|
||||
graph.addEdge(3, 5) // Component 2
|
||||
console.log(graph.connectedComponents())
|
||||
}
|
||||
|
||||
main()
|
Reference in New Issue
Block a user