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

示例 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

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;
    }
};