chal_53.py

class Solution(object):
    def max_subarray(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        max_sum = nums[0]
        curr_sum = 0

        for num in nums:
            if curr_sum < 0:
                curr_sum = 0
            curr_sum += num
            max_sum = max(max_sum, curr_sum)
        return max_sum


# Test cases
sol = Solution()

nums = [-2, 1, -3, 4, -1, 2, 1, -5, 4]
assert sol.max_subarray(nums) == 6

nums = [1]
assert sol.max_subarray(nums) == 1

nums = [5, 4, -1, 7, 8]
assert sol.max_subarray(nums) == 23