mirror of
https://github.com/TheAlgorithms/JavaScript.git
synced 2025-07-19 01:55:51 +08:00
41 lines
1.0 KiB
JavaScript
41 lines
1.0 KiB
JavaScript
// https://en.wikipedia.org/wiki/Neighbourhood_(graph_theory)
|
|
|
|
class Graph {
|
|
// Generic graph: the algorithm works regardless of direction or weight
|
|
constructor () {
|
|
this.edges = []
|
|
}
|
|
|
|
addEdge (node1, node2) {
|
|
// Adding edges to the graph
|
|
this.edges.push({
|
|
node1,
|
|
node2
|
|
})
|
|
}
|
|
|
|
nodeNeighbors (node) {
|
|
// Returns an array with all of the node neighbors
|
|
const neighbors = new Set()
|
|
for (let i = 0; i < this.edges.length; i++) {
|
|
// Checks if they have an edge between them and if the neighbor is not
|
|
// already in the neighbors array
|
|
if (this.edges[i].node1 === node && !(neighbors.includes(this.edges[i].node2))) {
|
|
neighbors.push(this.edges[i].node2)
|
|
} else if (this.edges[i].node2 === node && !(neighbors.includes(this.edges[i].node1))) {
|
|
neighbors.push(this.edges[i].node1)
|
|
}
|
|
}
|
|
return neighbors
|
|
}
|
|
}
|
|
|
|
(() => {
|
|
const graph = new Graph()
|
|
graph.addEdge(1, 2)
|
|
graph.addEdge(2, 3)
|
|
graph.addEdge(3, 5)
|
|
graph.addEdge(1, 5)
|
|
console.log(graph.nodeNeighbors(1))
|
|
})()
|