138.子数组之和

描述

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

注意事项

There is at least one subarray that it's sum equals to zero.

样例

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

知识点

所有子数组问题都需要求前缀和数组,数组的下标i代表原数组前i个数之和

代码

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<Integer> subarraySum(int[] nums) {
        ArrayList<Integer> ans = new ArrayList<Integer>();
        // map的key是子数组之和,value是当前位置
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        int sum = 0;
        // 数组内没有元素时,前缀和是0,若不加这行就会漏[0, 2]的解
        map.put(0, -1);
        
        for (int i = 0; i < nums.length; i++) {
            sum += nums[i];
            if (map.containsKey(sum)) {
                // prefix[i] 代表前 i 个数(下标 0 到 i-1)的和
                // prefix[j+1] - prefix[i] 表示从 i+1 到 j 的子数组
                // 对于 prefix[i] 要 +1, 因为子数组是从 i 的 下一个数开始的
                ans.add(map.get(sum) + 1);
                ans.add(i);
                return ans;
            }
            map.put(sum, i);
        }
        // 数组不存在或数组内没有元素时子数组和为0,不用写异常
        return ans;
    }
}

推荐阅读更多精彩内容