Leetcode 最长连续子数组

前言

  最近在做leetcode的时候，遇到了一类求最长子数组问题，它们的解法比较巧妙，特地记录一下。这两道题目分别是Maximum size subarray equals K和Contiguous array

Maximum size subarray equals K

  题目给定了一个数组，要求和为K的最长子数组的长度。为了求解这个题目，我们需要建立一个数组和到数组下标的映射，用到的是hashmap。为了能够快速找到区间长度，我们需要建立区间长度和，利用$S_i$表示下标0到i的数组和，那么区间[i,j]的数组和为$S_j-S_{i-1}$，该区间的长度为$j-i+1$。在hashmap中的元素为$<S[i],i>$这样的位置映射，那么，当我们遍历到第i位的时候，我们需要查看hashmap中的是否包含值为$S[i]-K$的键值，如果包含，那么用i减去相应的下标位置就是区间和为K的子数组。对应的代码如下：

int sumEqualsK(vector<int> nums,int K){
    map<int,int> m{{0,-1}};
    int sum=0;
    int res=0;
    for(int i=0;i<nums.size();i++){
        sum+=nums[i];
        if(m.count(sum-K)){
            res=max(res,i-m[sum-K]);
        }
        else{
            m[sum]=i;
        }
    }
    return res;
}

初始时map加入{0，-1},因为可能区间[0,i]的和为K，这样区间长度为i+1.

Contiguous Array

  这道题目给定了一个由0,1组成的数组，要求0，1数目相等的最长子数组的长度，这个题目和上一个题目思想比较相似，基本操作利用sum记录0，1的数目，遇到1就加1，遇到0就减1。利用map记录<sum[i],i>，当遍历到第i个元素时，查看map中是否包含sum[i]，如果包含，就更新当前值，如果不包含，就将sum[i]加入map中，代码如下：

    int findMaxLength(vector<int>& nums) {
        int n=nums.size();
        unordered_map<int,int> m{{0,-1}};
        int sum=0;
        int res=0;
        for(int i=0;i<n;i++){
            sum+=(nums[i]==1)?1:-1;
            if(m.count(sum)){
                res=max(res,i-m[sum]);
            }
            else{
                m[sum]=i;
            }
        }
        return res;
    }

总结

  这两道题目基本上是一样的，用到的思想也一样，解法比较巧妙。可惜做第二题的时候我没有想到第一题的思路。

参考资料：

• Subarray equals K
• Contiguous Array