@@ -102,24 +102,24 @@ def dfs(self, s=-2, d=-1):
102
102
stack = []
103
103
visited = []
104
104
if s == - 2 :
105
- if self .graph .get (s ,None ):
106
- pass # -2 is a node
105
+ if self .graph .get (s , None ):
106
+ pass # -2 is a node
107
107
elif len (self .graph ) > 0 :
108
108
s = next (iter (self .graph ))
109
109
else :
110
- return [] # Graph empty
110
+ return [] # Graph empty
111
111
stack .append (s )
112
112
113
113
# Run dfs
114
114
while len (stack ) > 0 :
115
115
s = stack .pop ()
116
116
visited .append (s )
117
117
# If reached d, return
118
- if s == d :
118
+ if s == d :
119
119
break
120
120
121
121
# add not visited child nodes to stack
122
- for _ ,ss in self .graph [s ]:
122
+ for _ , ss in self .graph [s ]:
123
123
if visited .count (ss ) < 1 :
124
124
stack .append (ss )
125
125
return visited
@@ -164,12 +164,12 @@ def bfs(self, s=-2):
164
164
d = deque ()
165
165
visited = []
166
166
if s == - 2 :
167
- if self .graph .get (s ,None ):
168
- pass # -2 is a node
167
+ if self .graph .get (s , None ):
168
+ pass # -2 is a node
169
169
elif len (self .graph ) > 0 :
170
170
s = next (iter (self .graph ))
171
171
else :
172
- return [] # Graph empty
172
+ return [] # Graph empty
173
173
d .append (s )
174
174
visited .append (s )
175
175
# Run bfs
@@ -413,12 +413,12 @@ def dfs(self, s=-2, d=-1):
413
413
stack = []
414
414
visited = []
415
415
if s == - 2 :
416
- if self .graph .get (s ,None ):
417
- pass # -2 is a node
416
+ if self .graph .get (s , None ):
417
+ pass # -2 is a node
418
418
elif len (self .graph ) > 0 :
419
419
s = next (iter (self .graph ))
420
420
else :
421
- return [] # Graph empty
421
+ return [] # Graph empty
422
422
stack .append (s )
423
423
424
424
# Run dfs
@@ -429,11 +429,11 @@ def dfs(self, s=-2, d=-1):
429
429
else :
430
430
visited .append (s )
431
431
# If reached d, return
432
- if s == d :
432
+ if s == d :
433
433
break
434
434
435
435
# add not visited child nodes to stack
436
- for _ ,ss in self .graph [s ]:
436
+ for _ , ss in self .graph [s ]:
437
437
if visited .count (ss ) < 1 :
438
438
stack .append (ss )
439
439
return visited
@@ -478,12 +478,12 @@ def bfs(self, s=-2):
478
478
d = deque ()
479
479
visited = []
480
480
if s == - 2 :
481
- if self .graph .get (s ,None ):
482
- pass # -2 is a node
481
+ if self .graph .get (s , None ):
482
+ pass # -2 is a node
483
483
elif len (self .graph ) > 0 :
484
484
s = next (iter (self .graph ))
485
485
else :
486
- return [] # Graph empty
486
+ return [] # Graph empty
487
487
d .append (s )
488
488
visited .append (s )
489
489
while d :
0 commit comments