Create minimum-number-of-operations-to-move-all-balls-to-each-box.py

Author: kamyu104

Python/minimum-number-of-operations-to-move-all-balls-to-each-box.py

@@ -0,0 +1,18 @@
+# Time:  O(n)
+# Space: O(1)
+
+class Solution(object):
+    def minOperations(self, boxes):
+        """
+        :type boxes: str
+        :rtype: List[int]
+        """
+        result = [0]*len(boxes)
+        for direction in (lambda x:x, reversed):
+            cnt = accu = 0
+            for i in direction(xrange(len(boxes))):
+                result[i] += accu
+                if boxes[i] == '1':
+                    cnt += 1
+                accu += cnt
+        return result