3432. Count Partitions with Even Sum Difference

3432. Count Partitions with Even Sum Difference

class Solution:
    def countPartitions(self, nums: List[int]) -> int:
        total = sum(nums)
        prefixSum = [num for num in nums]
        for i in range(1, len(nums)):
            prefixSum[i] = prefixSum[i - 1] + nums[i]
        
        count = 0
        for i in range(len(prefixSum) - 1):
            left = prefixSum[i]
            right = total - left
            if abs(left - right) % 2 == 0:
                count += 1
        
        return count