LeetCode 525. 连续数组（前缀和+哈希）

1. 题目

给定一个二进制数组, 找到含有相同数量的 0 和 1 的最长连续子数组（的长度）。

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/contiguous-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

类似题目：LeetCode 560. 和为K的子数组（前缀和差分）

• 把0看成-1，找和为0的最长长度，跟560题差不多
• 记录前缀和，第一次出现时，存入map，记录下标
• 第二次出现时，做差，中间的长度就是和为0的区间

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
    	int i, n = nums.size(), sum = 0, maxlen = 0;
    	unordered_map<int,int> m;//sum, 第一次出现sum时的idx
        m[0] = -1;
    	for(i = 0; i < n; ++i)
    	{
    		if(nums[i])
    			sum += 1;
    		else
    			sum -= 1;
    		if(!m.count(sum))
    			m[sum] = i;
    		else
    			maxlen = max(maxlen, i-m[sum]);
    	}
    	return maxlen;
    }
};

344 ms 78.5 MB