凡客网站设计/长沙企业seo优化

题目

给你一个整数数组 nums 和一个整数 k ，请你统计并返回该数组中和为 k 的连续子数组的个数。

示例

输入：nums = [1,1,1], k = 2
输出：2

输入：nums = [1,2,3], k = 3
输出：2

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarray-sum-equals-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：前缀和

1、preSum[i] 记录 nums[0…i-1] 的累加和。
2、如果我想求索引区间 [1, 4] 内的所有元素之和，就可以通过 preSum[5] - preSum[1] 得出。

Java实现1

class Solution {public int subarraySum(int[] nums, int k) {int n = nums.length;int[] preSum = new int[n + 1];preSum[0] = 0;int flag = 0;for (int i = 0; i < n; i++){preSum[i + 1] = preSum[i] + nums[i];}for (int i = 0; i < n + 1; i++){for (int j = i + 1; j < n + 1; j++){if (preSum[j] - preSum[i] == k) flag++;}}return flag;}
}

方法2：前缀和+哈希表

Java实现

class Solution {public int subarraySum(int[] nums, int k) {int n = nums.length, ans = 0;int[] preSum = new int[n + 1];preSum[0] = 0;for (int i = 0; i < n; i++) {preSum[i + 1] = preSum[i] + nums[i];}Map<Integer, Integer> map = new HashMap<>();map.put(0, 1);for (int i = 1; i <= n; i++) {int t = preSum[i];int d = t - k;ans += map.getOrDefault(d, 0);map.put(t, map.getOrDefault(t, 0) + 1);}return ans;}
}