152、和为K的子数组

最新推荐文章于 2025-06-01 23:23:20 发布

hello kitty w

最新推荐文章于 2025-06-01 23:23:20 发布

阅读量138

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode中等

本文链接：https://blog.youkuaiyun.com/qq_34446716/article/details/95042685

LeetCode中等专栏收录该内容

280 篇文章

订阅专栏

本文探讨了在给定数组中寻找和为特定值K的连续子数组的方法。介绍了两种算法实现，一种通过计算前缀和并进行两层遍历，另一种利用哈希表优化查找过程，显著提高效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：
给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明 :

数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

先将前i的和放入到sum数组中，然后针对每个索引区间进行计算，这时候求和就直接减去就可以了

class Solution {
   public int subarraySum(int[] nums, int k) {
		int time = 0;
		int []sum = Arrays.copyOf(nums,nums.length);
		for (int i = 1; i < sum.length; i++) {
			sum[i] = sum[i - 1] + sum[i];
		}
		for (int i = 0; i < nums.length; i++) {
			for (int j = i; j < sum.length; j++) {
				if(get1(sum, i, j) == k){
					time ++;
				}
			}
		}
		
		return time;
        
    }
	public int get1(int sum[],int start,int end){
		if(start == 0){
			return sum[end];
		}else {
			return sum[end] - sum[start - 1];
		}
	}
	
}

换个思路：
使用Hash表，需要注意的是初始化的put(0,1)，因为如果sum为k那么表示从开始的和就是k，因此初始化为1，接下来我们只需要更新map并且计算sum - k在表中出现的次数。

		Map<Integer, Integer> map = new HashMap<>();
		int time = 0;
		int sum = 0;
		map.put(0, 1);
		for (int i : nums) {
			sum += i;
			time += map.getOrDefault(sum - k, 0);
			map.put(sum, map.getOrDefault(sum, 0) + 1);
		}
		return time;

速度明显提升。