题目描述
给定一个无序数组arr, 其中元素可正、可负、可0。给定一个整数k,求arr所有子数组中累加和为k的最长子数组长度。
示例1
输入 [1,-2,1,1,1],0
返回值 3
题解:
问题中给定的是子数组,可以想到使用前缀和,然后将前缀和以及下标存储到hash表中,如果前缀和恰好等于k,比较一下当前k的位置与此前记录的k的值的大小取最大值。如果是区间[ l,r ]和等于k,其长度等于s[r] - s[l-1] == k ,要想求解s[l-1] 是否在hash中存在,可以直接使用hash.find判断s[l-1] 也就是 s[r] - k是否存在。
注意: hash中已经存在的值就不要在加入了,这样保存的是最前面的位置,也就是不要使用hash[s[i]]=i;来存储前缀和信息。
class Solution {
public:
int maxlenEqualK(vector<int>& arr, int k) {
// write code here
int n=arr.size();
int res=0;
vector<int> s(n+1,0);
unordered_map<int,int> hash;
for(int i=1;i<=n;i++)
{
s[i]=s[i-1]+arr[i-1];
if(hash.find(s[i])==hash.end())
{
hash.insert({s[i],i});
}
if(s[i]==k)
{
res=max(res,i);
}
if(hash.find(s[i]-k)!=hash.end())
{
res=max(i-hash[s[i]-k],res);
}
}
return res;
}
};