题目
方法一:
构造一个前缀和数组,然后在两重循环暴力搜索所有的可能;时间复杂度O(n^2);最后一个测试用例无法通过。
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
vector<long long>sum{0};
long long ans=0,res=0;
for(auto c:nums)
{
ans+=c;
sum.push_back(ans);
}
for(int i=1;i<sum.size();++i)
{
for(int j=i;j<sum.size();++j)
{
auto val=sum[j]-sum[i-1];
if(val==k)++res;
}
}
return res;
}
};
方法二:
利用一个哈希表保存前缀和 (即<前缀和,个数>),时间复杂度为O(n);
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int,int>ma;
int cur=0,cnt=0;
ma[0]=1;
for(int i=0;i<nums.size();++i)
{
cur+=nums[i];
auto ret=ma.find(cur-k);
if(ret!=ma.end())cnt+=ret->second;
ma[cur]++;
}
return cnt;
}
};
总结:1. 用哈希表来保存前缀和应该是经常用到的方法
2. 如果数组元素全为正数或者负数,那么可以考虑使用双指针的方法;
如果数组元素都为正数,前缀和数组有时候可能与二分法结合使用。