560. 和为 K 的子数组
1、题目
给你一个整数数组 nums
和一个整数 k
,请你统计并返回 该数组中和为 k
的子数组的个数 。
子数组是数组中元素的连续非空序列。
示例 1:
输入:nums = [1,1,1], k = 2 输出:2
示例 2:
输入:nums = [1,2,3], k = 3 输出:2
提示:
1 <= nums.length <= 2 * 104
-1000 <= nums[i] <= 1000
-107 <= k <= 107
Related Topics
- 数组
- 哈希表
- 前缀和
2、题目分析
1.求两个位置之间的元素和。可以转换成 两个位置前缀和之间的差值
2.A+B=k,可以转换成计算 B-k=A。(无后效性)
3、复杂度最优解代码示例
public int subarraySum(int[] nums, int k) {
// 用于记录前缀和及其出现次数的哈希表
Map<Integer, Integer> map = new HashMap<>();
map.put(0, 1); // 初始时,前缀和为0的出现次数为1
int count = 0;
int sum = 0;
for (int num : nums) {
sum += num; // 计算当前位置的前缀和
if (map.containsKey(sum - k)) {
// 更新计数。如果存在前缀和为sum - k的位置,则说明存在和为k的子数组
count += map.get(sum - k);
}
// 更新当前前缀和的出现次数
map.put(sum, map.getOrDefault(sum, 0) + 1);
}
return count;
}
4、扩展-生活应用场景
如投资管理:
假设你是一位投资经理,负责管理一个投资组合,该投资组合包含多种资产,如股票、债券、期货等。你的目标是通过合理的投资组合来最大化投资回报,并控制投资组合的风险。
在这个场景中,你需要考虑各种投资组合的收益和风险。你可能会设定一个目标收益 K,希望投资组合在一定时间内达到或超过这个目标收益。但是,由于市场的波动和资产之间的相关性,实际的投资回报可能会有所偏差。
"和为 K 的子数组"问题可以类比为投资管理中的情况。你可以将投资组合的收益看作是一个数组,数组中的元素表示各种不同投资组合的收益率。你需要找到投资组合中连续的子数组,使得这些子数组的收益之和等于目标收益 K。
通过解决"和为 K 的子数组"问题,你可以找到满足目标收益的投资组合,并进行相应的调整和优化,以实现投资管理的目标。