Create all-divisions-with-the-highest-score-of-a-binary-array.py

Author: kamyu104

Python/all-divisions-with-the-highest-score-of-a-binary-array.py

@@ -0,0 +1,21 @@
+# Time:  O(n)
+# Space: O(1)
+
+# prefix sum
+class Solution(object):
+    def maxScoreIndices(self, nums):
+        """
+        :type nums: List[int]
+        :rtype: List[int]
+        """
+        result = []
+        mx = zeros = 0
+        total = sum(nums)
+        for i in xrange(len(nums)+1):
+            zeros += ((nums[i-1] if i else 0) == 0)
+            if zeros+(total-(i-zeros)) > mx:
+                mx = zeros+(total-(i-zeros))
+                result = []
+            if zeros+(total-(i-zeros)) == mx:
+                result.append(i)
+        return result