LeetCode 325. Maximum Size Subarray Sum Equals k / 525. Contiguous Array

325. Maximum Size Subarray Sum Equals k

类似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;
    }
};

 

525. Contiguous Array

把0看做-1，那么就变成一个求累计sum=0的问题了，和上面的sum=k的问题一模一样。

可以通过加上 hash[0]=-1 把 累计sum==0的 情况一起并进去。

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        int maxlen=0;
        int sum=0;
        unordered_map<int,int> hash; //sum, index
        hash.insert({0,-1});
        for (int end=0;end<nums.size();++end){
            sum += nums[end]==0?-1:1;
            // find sum[end]-end[i]=0 so that [i+1,end] has equal number of 0 and 1
            if (hash.count(sum)){
                maxlen = max(maxlen, end-hash[sum]);
            }else{
                hash[sum] = end;
            }
        }
        return maxlen;
    }
};