LeetCode 325. Maximum Size Subarray Sum Equals k

Posted 2021-01-04 約束の空

类似sum的题都可以想一想可不可以把sum依次求出来，然后 [i+1,j] 的sum 直接用 sum[j]-sum[i]来求。

由于这道题有负数存在，没办法用sliding window来做，只能另辟蹊径。

用一个hashtable保存到目前为止的sum所对应的index，这样之后知道寻找 sum-k 在不在hashtable里就行了。如果遇到sum相同的情况，不要更新hashtable，因为更新了，subarray的size必然是减小的。

class Solution {
public:
    int maxSubArrayLen(vector<int>& nums, int k) {
        int res=0;
        unordered_map<int,int> hash; //sum, index
        int sum=0;
        for (int i=0;i<nums.size();++i){
            sum += nums[i];
            if (sum==k) res=max(res,i-0+1);
            if (hash.count(sum-k)) res=max(res,i-hash[sum-k]); // [hash[sum-k]+1,i] sum up to k
            if (!hash.count(sum)) hash[sum]=i;
        }
        return res;
    }
};