aboutsummaryrefslogtreecommitdiff
path: root/maximum_subarray.py
diff options
context:
space:
mode:
Diffstat (limited to 'maximum_subarray.py')
-rw-r--r--maximum_subarray.py29
1 files changed, 29 insertions, 0 deletions
diff --git a/maximum_subarray.py b/maximum_subarray.py
new file mode 100644
index 0000000..1edbefc
--- /dev/null
+++ b/maximum_subarray.py
@@ -0,0 +1,29 @@
+# 53. Maximum Subarray
+
+"""
+given an integer array 'nums', find the subarray with the largest sum, and
+return its sum.
+"""
+
+
+class Solution(object):
+ def maxSubArray(self, nums):
+ """
+ :type nums: List[int]
+ :rtype: int
+ """
+ max_sum, curr_sum = float("-inf"), 0
+ for i in nums:
+ curr_sum += i
+ if max_sum < curr_sum:
+ max_sum = curr_sum
+ if curr_sum < 0:
+ curr_sum = 0
+ return max_sum
+
+
+if __name__ == "__main__":
+ obj = Solution()
+ print(obj.maxSubArray(nums=[-2, 1, -3, 4, -1, 2, 1, -5, 4]))
+ print(obj.maxSubArray(nums=[1]))
+ print(obj.maxSubArray(nums=[5, 4, -1, 7, 8]))