和为 K 的子数组(数组技巧,map,leetcode560)-------------------c++实现
题目表述
给你一个整数数组 nums 和一个整数 k ,请你统计并返回 该数组中和为 k 的子数组的个数 。
样例
输入:nums = [1,1,1], k = 2
输出:2
输入:nums = [1,2,3], k = 3
输出:2
条件
1 <= nums.length <= 2 * 104
-1000 <= nums[i] <= 1000
-107 <= k <= 107
思路
计算前n项和依次放入map中,通过计算当前前now项和与target值的差值,然后看前面的now-1项中前n项和中为该值的有多少个,用map数据结构记录前n项和为key的value值(即重复个数).
注意点
注意map值的应用
ac代码
c++:
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int,int> mp;
mp[0]=1; //为了看当前值是不是目标值
int result =0,sum=0;
for(auto &x:nums) //for (vector< int >::iterator iter = nums.begin(); iter != nums.end(); iter++)
{
sum+=x;
if(mp.count(sum-k))
result+=mp[sum-k];
mp[sum]++;
}
return result;
}
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/squares-of-a-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。