CDE

Task/Equilibrium-index/Python/equilibrium-index-1.py

@@ -0,0 +1,9 @@
+def eqindex2Pass(data):
+    "Two pass"
+    suml, sumr, ddelayed = 0, sum(data), 0
+    for i, d in enumerate(data):
+        suml += ddelayed
+        sumr -= d
+        ddelayed = d
+        if suml == sumr:
+            yield i

Task/Equilibrium-index/Python/equilibrium-index-2.py

@@ -0,0 +1,6 @@
+def eqindexMultiPass(data):
+    "Multi pass"
+    for i in range(len(data)):
+        suml, sumr = sum(data[:i]), sum(data[i+1:])
+        if suml == sumr:
+            yield i

Task/Equilibrium-index/Python/equilibrium-index-3.py

@@ -0,0 +1,4 @@
+def eqindexMultiPass(s):
+    return [i for i in xrange(len(s)) if sum(s[:i]) == sum(s[i+1:])]
+
+print eqindexMultiPass([-7, 1, 5, 2, -4, 3, 0])

Task/Equilibrium-index/Python/equilibrium-index-4.py

@@ -0,0 +1,9 @@
+from collections import defaultdict
+
+def eqindex1Pass(data):
+    "One pass"
+    l, h = 0, defaultdict(list)
+    for i, c in enumerate(data):
+        l += c
+        h[l * 2 - c].append(i)
+    return h[l]

Task/Equilibrium-index/Python/equilibrium-index-5.py

@@ -0,0 +1,10 @@
+f = (eqindex2Pass, eqindexMultiPass, eqindex1Pass)
+d = ([-7, 1, 5, 2, -4, 3, 0],
+     [2, 4, 6],
+     [2, 9, 2],
+     [1, -1, 1, -1, 1, -1, 1])
+
+for data in d:
+    print("d = %r" % data)
+    for func in f:
+        print("  %16s(d) -> %r" % (func.__name__, list(func(data))))