华为面试题：和为K的子数组

最新推荐文章于 2024-11-03 21:27:26 发布

四库全书的酷

最新推荐文章于 2024-11-03 21:27:26 发布

阅读量166

点赞数

分类专栏：刷题cpp 文章标签：算法哈希算法

本文链接：https://blog.csdn.net/weixin_46535567/article/details/125166489

版权

刷题cpp 专栏收录该内容

23 篇文章 4 订阅

订阅专栏

本文介绍了如何解决在一个无序整数数组中，找到和为特定目标K的子数组个数的问题。通过结合前缀和（prefix sum）技巧和哈希表（unordered_map），提供了一段高效的算法实现。适用于正负混合且无序的数组场景。

摘要由CSDN通过智能技术生成

问题

求一个数组Vector中的某个子数组所有元素和为K的个数；

分析

情况一：

数组Vector中的元素都是正数或者都是负数的话，可以使用滑动指针的方法

情况二：

数组元素是无序的，可以为正负任意的。

此时可以使用通用方法：前缀和+HASH

int func(vector<int>& v,int target){
	unordered_map<int,int> cap;
	cap[0] = 1;
	int pre = 0;
	int cnt = 0;
	for(int i = 0; i < v.size(); ++i){
		pre += v[i];
		if(cap.count(pre - target)){
			cnt += cap[pre - target];
		}
		cap[pre]++;
	}
	return cnt;
}