525. 连续数组

给定一个二进制数组 nums , 找到含有相同数量的 0 和 1 的最长连续子数组，并返回该子数组的长度。

 

示例 1:

输入: nums = [0,1]
输出: 2
说明: [0, 1] 是具有相同数量0和1的最长连续子数组。

示例 2:

输入: nums = [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。

 

提示：

• 1 <= nums.length <= 105
• nums[i] 不是 0 就是 1

 

map保存prefix==0的{key,value},求最大ans

 

c++

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        int ans=0,sum=0,n=nums.size();
        map<int,int> m;
        for(int i=0;i<n;i++){
            sum+=nums[i]==0?-1:1;
            if(sum==0){
                if(i>ans)ans=i+1;
            }
            if(m.find(sum)==m.end()){
                m.insert({sum,i});
                continue;
            }
            ans=max(ans,i-m.find(sum)->second);
        }
        return ans;
    }
};

 

Java

class Solution {
    public int findMaxLength(int[] nums) {
        int ans=0,sum=0;
        int n=nums.length;
        for(int i=0;i<n;i++){
            if(nums[i]==0)nums[i]=-1;
        }
        HashMap<Integer,Integer> map = new HashMap();
        for(int i=0;i<n;i++){
            sum+=nums[i];
            if(sum==0){
                if(i>ans)ans=i+1;
            }
            if(map.get(sum)==null){
                map.put(sum,i);
                continue;
            }
            ans=Math.max(ans,i-map.get(sum));
        }
        return ans;
    }
}