File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -17,13 +17,13 @@ class Graph {
17
17
nodeNeighbors ( node ) {
18
18
// Returns an array with all of the node neighbors
19
19
const neighbors = new Set ( )
20
- for ( let i = 0 ; i < this . edges . length ; i ++ ) {
20
+ for ( let edge of this . edges ) {
21
21
// Checks if they have an edge between them and if the neighbor is not
22
22
// 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 )
27
27
}
28
28
}
29
29
return neighbors
You can’t perform that action at this time.
0 commit comments