C ++中的子数组总和等于K

假设我们有一个整数数组和一个整数k，我们需要找到总和与k相等的连续子数组的总数。因此，如果nums array为[1，1，1]并且k为2，则输出将为2。

为了解决这个问题，我们将遵循以下步骤-

• 定义一个称为sums的映射，temp：= 0，sums [0]：= 1和ans：= 0

• 对于范围在0到数组大小的i

• ans：= ans + sums [k-temp]

• 温度：=温度+ n [i]

• 如果总和为k – temp，则

• 将sums [-temp]的值增加1

• 返回ans

例子（C ++）

让我们看下面的实现以更好地理解-

#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
   int subarraySum(vector<int>& n, int k) {
      unordered_map <int, int> sums;
      int temp = 0;
      sums[0] = 1;
      int ans =0;
      for(int i =0;i<n.size();i++){
         temp+= n[i];
         if(sums.find(k-temp)!=sums.end()){
            ans += sums[k-temp];
         }
         sums[-temp]++;
      }
      return ans;
   }
};
main(){
   Solution ob;
   vector<int> v = {1,1,1};
   cout << (ob.subarraySum(v, 2));
}

输入值

[1,1,1]
2

输出结果

2