Skip to content

Commit 686ba7e

Browse files
committed
Added suggestions and updated for loop
1 parent 1d72cbb commit 686ba7e

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

Graphs/NodeNeighbors.js

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,13 +17,13 @@ class Graph {
1717
nodeNeighbors (node) {
1818
// Returns an array with all of the node neighbors
1919
const neighbors = new Set()
20-
for (let i = 0; i < this.edges.length; i++) {
20+
for (let edge of this.edges) {
2121
// Checks if they have an edge between them and if the neighbor is not
2222
// already in the neighbors array
23-
if (this.edges[i].node1 === node && !(neighbors.includes(this.edges[i].node2))) {
24-
neighbors.push(this.edges[i].node2)
25-
} else if (this.edges[i].node2 === node && !(neighbors.includes(this.edges[i].node1))) {
26-
neighbors.push(this.edges[i].node1)
23+
if (edge.node1 === node && !(neighbors.has(edge.node2))) {
24+
neighbors.add(edge.node2)
25+
} else if (edge.node2 === node && !(neighbors.has(edge.node1))) {
26+
neighbors.add(edge.node1)
2727
}
2828
}
2929
return neighbors

0 commit comments

Comments
 (0)