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. 如果數組元素全爲正數或者負數，那麼可以考慮使用雙指針的方法；
如果數組元素都爲正數，前綴和數組有時候可能與二分法結合使用。