Create greatest-sum-divisible-by-three.py

Author: kamyu104

Python/greatest-sum-divisible-by-three.py

@@ -0,0 +1,14 @@
+# Time:  O(n)
+# Space: O(1)
+
+class Solution(object):
+    def maxSumDivThree(self, nums):
+        """
+        :type nums: List[int]
+        :rtype: int
+        """
+        dp = [0, 0, 0]
+        for num in nums:
+            for i in [num+x for x in dp]:
+                dp[i%3] = max(dp[i%3], i)
+        return dp[0]