【leetcode刷题】T50-连续数组

Input: [,]
Output: 
Explanation: [, ] is the longest contiguous subarray with equal number of  and .

Input: [,,]
Output: 
Explanation: [, ] (or [, ]) is a longest contiguous subarray with equal number of  and .

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

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

class Solution(object):
    def findMaxLength(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        res = 
        d = {:-1}
        sum0 = 
        for i, n in enumerate(nums):
            if n == :
                sum0 -= 
            else:
                sum0 += 
            if sum0 in d:
                res = max(res, i - d[sum0])
            else:
                d[sum0] = i
        return res

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        int count = ;
        int res = ;
        map<int, int> d;
        d[] = -1;
        for(int i=; i<nums.size(); i++){
            if(nums[i] == )
                count++;
            else
                count--;
            if(d.find(count) == d.end())
                d[count] = i;
            else
                res = max(res, i - d[count]);
        }
        return res;
    }
};