LintCode 子数组之和

题目

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

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

分析

这里用了一个技巧：将数组从第一位依次相加，记录每次的结果，如果map里面没有，就加入map里，如果有，就证明前面肯定有为0的子数组，才会出现一样的和。

代码

public 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
     */
    public ArrayList subarraySum(int[] nums) {
        // write your code here
        int len = nums.length;
       
        ArrayList ans = new ArrayList();
        HashMap map = new HashMap();
       
        map.put(0, -1);
       
        int sum = 0;
        for (int i = 0; i < len; i++) {
            sum += nums[i];
           
            if (map.containsKey(sum)) {
                ans.add(map.get(sum) + 1);
                ans.add(i);
                return ans;
            }
            
            map.put(sum, i);
        }
       
        return ans;
    }
}

.