Create partition-array-into-three-parts-with-equal-sum.py

Author: kamyu104

Python/partition-array-into-three-parts-with-equal-sum.py

@@ -0,0 +1,19 @@
+# Time:  O(n)
+# Space: O(1)
+
+class Solution(object):
+    def canThreePartsEqualSum(self, A):
+        """
+        :type A: List[int]
+        :rtype: bool
+        """
+        total = sum(A)
+        if total % 3 != 0:
+            return False
+        parts, curr = 0, 0
+        for x in A:
+            curr += x
+            if curr == total//3:
+                parts += 1
+                curr = 0
+        return parts >= 3