C ++中的子数组总和等于K
假设我们有一个整数数组和一个整数k,我们需要找到总和与k相等的连续子数组的总数。因此,如果numsarray为[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