各个元素之和等于给定值的连续子数组的个数

Posted 2020-12-14 poser

给定一个数组arr，给定一个数target。求，各个元素之和等于target的连续子数组的个数，时间复杂度为O(N)，空间复杂度也为O(N)

借助一个Map即可实现。Map中保存的是：从arr的最左侧元素（也就是下标为0的元素）开始，到当前位置为止，所有元素的和出现的次数。

 1 private int getNumOfSubThatSumIsTarget(int[] arr, int target) {
 2     Map<Integer, Integer> map = new HashMap<>();
 3     map.put(0, 1);
 4     
 5     int result = 0;
 6     int curSum = 0;
 7     for (int i = 0; i < arr.length; ++i) {
 8         curSum += arr[i];
 9         
10         result += map.getOrDefault(curSum - target, 0);
11         
12         map.put(curSum, map.getOrDefault(curSum, 0) + 1);
13     }
14     
15     return result;
16 }