前缀和思想:
为1时,count = 1; 为0时,count = -1;
初始化map[0] = -1
遍历数组,得到前缀和sum
map记录不同前缀和首次出现的位置
i - map[sum] 即为相同前缀和之间的长度
result记录最长连续子数组的长度
class Solution {
public:
int findMaxLength(vector<int>& nums) {
unordered_map<int,int> map;
int count = 0;
map[count] = -1;
int result = 0;
int sum = 0;
for(int i = 0; i < nums.size();i++)
{
sum += ( nums[i] == 1 ) ? 1 : -1;
if(map.count(sum))
{
result = max(result, i - map[sum]);
}
else
map[sum] = i;
}
return result;
}
};