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. 如果数组元素全为正数或者负数,那么可以考虑使用双指针的方法;
如果数组元素都为正数,前缀和数组有时候可能与二分法结合使用。

你可能感兴趣的:(算法,LeetCode)