[LintCode 138] 子数组之和(Python)

题目描述

给定一个整数数组，找到和为零的子数组。你的代码应该返回满足要求的子数组的起始位置和结束位置

注意事项
There is at least one subarray that it’s sum equals to zero.

样例
给出 [-3, 1, 2, -3, 4]，返回[0, 2] 或者 [1, 3].

思路

用字典存储中间结果。key为从第一个元素到当前元素的和，value为下标。当某个key值第二次出现时，则它们中间的子数组和一定为0。

代码

class Solution:
    """
    @param nums: A list of integers
    @return: A list of integers includes the index of the first number
             and the index of the last number
    """
    def subarraySum(self, nums):
        # write your code here
        res = []
        if nums is None or len(nums) <= 0:
            return res
        d = {0: -1}
        sum = 0
        for i in range(len(nums)):
            sum += nums[i]
            if sum in d:
                res.append(i)
                res.append(d[sum] + 1)
                return res
            d[sum] = i
        return res

复杂度分析

时间复杂度$O(n)$，空间复杂度$O(n)$。