leetcode0525. 连续数组-medium

1 题目：连续数组

官方标定难度：中

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

示例 1：

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

示例 2：

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

示例 3：

输入：nums = [0,1,1,1,1,1,0,0,0]
输出：6
解释：[1,1,1,0,0,0] 是具有相同数量 0 和 1 的最长连续子数组。

提示：

1 <= nums.length <= $10^5$
nums[i] 不是 0 就是 1

2 solution

代码

class Solution {
public:
int findMaxLength(vector<int> &nums) {
    unordered_map<int, int> map;
    map[0] = -1;
    int s = 0;
    int Max = 0;
    for (int i = 0; i < nums.size(); i++) {
        s += nums[i] * 2 - 1; // 映射成 -1 和 1；
        // cout << s << " ";
        if (map.find(s) != map.end()) {
            Max = max(Max, i - map[s]);
            //cout << i - map[s];
        }else{
            map[s] = i;
        }
    }
    return Max;
}
};

结果