Leetcode 560和为K的子数组

题目
方法一:
构造一个前缀和数组,然后在两重循环暴力搜索所有的可能;时间复杂度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. 如果数组元素全为正数或者负数,那么可以考虑使用双指针的方法;
如果数组元素都为正数,前缀和数组有时候可能与二分法结合使用。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值