和为 K 的子数组(数组技巧,map,leetcode560)-------------------c++实现

和为 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

你可能感兴趣的:(c++,leetcode,算法)