@@ -39,7 +39,7 @@ def dfs(self, s=-2, d=-1):
39
39
stack = []
40
40
visited = []
41
41
if s == - 2 :
42
- s = list ( self .graph )[ 0 ]
42
+ s = next ( iter ( self .graph ))
43
43
stack .append (s )
44
44
visited .append (s )
45
45
ss = s
@@ -87,7 +87,7 @@ def bfs(self, s=-2):
87
87
d = deque ()
88
88
visited = []
89
89
if s == - 2 :
90
- s = list ( self .graph )[ 0 ]
90
+ s = next ( iter ( self .graph ))
91
91
d .append (s )
92
92
visited .append (s )
93
93
while d :
@@ -114,7 +114,7 @@ def topological_sort(self, s=-2):
114
114
stack = []
115
115
visited = []
116
116
if s == - 2 :
117
- s = list ( self .graph )[ 0 ]
117
+ s = next ( iter ( self .graph ))
118
118
stack .append (s )
119
119
visited .append (s )
120
120
ss = s
@@ -146,7 +146,7 @@ def topological_sort(self, s=-2):
146
146
def cycle_nodes (self ):
147
147
stack = []
148
148
visited = []
149
- s = list ( self .graph )[ 0 ]
149
+ s = next ( iter ( self .graph ))
150
150
stack .append (s )
151
151
visited .append (s )
152
152
parent = - 2
@@ -199,7 +199,7 @@ def cycle_nodes(self):
199
199
def has_cycle (self ):
200
200
stack = []
201
201
visited = []
202
- s = list ( self .graph )[ 0 ]
202
+ s = next ( iter ( self .graph ))
203
203
stack .append (s )
204
204
visited .append (s )
205
205
parent = - 2
@@ -305,7 +305,7 @@ def dfs(self, s=-2, d=-1):
305
305
stack = []
306
306
visited = []
307
307
if s == - 2 :
308
- s = list ( self .graph )[ 0 ]
308
+ s = next ( iter ( self .graph ))
309
309
stack .append (s )
310
310
visited .append (s )
311
311
ss = s
@@ -353,7 +353,7 @@ def bfs(self, s=-2):
353
353
d = deque ()
354
354
visited = []
355
355
if s == - 2 :
356
- s = list ( self .graph )[ 0 ]
356
+ s = next ( iter ( self .graph ))
357
357
d .append (s )
358
358
visited .append (s )
359
359
while d :
@@ -371,7 +371,7 @@ def degree(self, u):
371
371
def cycle_nodes (self ):
372
372
stack = []
373
373
visited = []
374
- s = list ( self .graph )[ 0 ]
374
+ s = next ( iter ( self .graph ))
375
375
stack .append (s )
376
376
visited .append (s )
377
377
parent = - 2
@@ -424,7 +424,7 @@ def cycle_nodes(self):
424
424
def has_cycle (self ):
425
425
stack = []
426
426
visited = []
427
- s = list ( self .graph )[ 0 ]
427
+ s = next ( iter ( self .graph ))
428
428
stack .append (s )
429
429
visited .append (s )
430
430
parent = - 2
0 commit comments