Update maximize-value-of-function-in-a-ball-passing-game.py

kamyu104 · web-flow · commit 19f90fb21c34 · 2023-08-28T17:27:08.000+08:00
diff --git a/Python/maximize-value-of-function-in-a-ball-passing-game.py b/Python/maximize-value-of-function-in-a-ball-passing-game.py
@@ -14,15 +14,16 @@ def getMaxFunctionValue(self, receiver, k):
         """
         def find_cycles(adj):
             result = []
-            lookup = [False]*len(adj)
+            lookup = [0]*len(adj)
+            idx = 0
             for u in xrange(len(adj)):
-                cnt = {}
+                prev = idx
                 while not lookup[u]:
-                    lookup[u] = True
-                    cnt[u] = len(cnt)
+                    idx += 1
+                    lookup[u] = idx
                     u = adj[u]
-                if u in cnt:
-                    result.append((u, len(cnt)-cnt[u]))
+                if lookup[u] > prev:
+                    result.append((u, idx-lookup[u]+1))
             return result
 
         def find_prefixes():
