当前位置 博文首页 > Michael是个半路程序员：LeetCode 525. 连续数组（前缀和+哈希）
给定一个二进制数组, 找到含有相同数量的 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
类似题目：
LeetCode 560. 和为K的子数组（前缀和差分）
LeetCode 1248. 统计「优美子数组」（要复习）
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
cs