Leetcode 525. 连续数组 C++

Leetcode 525. 连续数组

题目

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

测试样例

示例 1:

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

示例 2:

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

注意: 给定的二进制数组的长度不会超过50000。

题解

我们转化一下，我们将0该为1，则我们的目标就是找最长连续数组和为0的数组。
利用前缀和进行求解
我们用哈希表记录最早出现的前缀和的下标，对于当前的前缀和，如果存在负和，则说明有一段连续数组和为0，我们更新答案。详细过程见代码

代码

	int findMaxLength(vector<int>& nums) {
        unordered_map<int,int> preSum;
        int n = nums.size();
        int sum=0,ans=0;
        for(int i=0; i<n; i++){
            if(nums[i]==0)  nums[i] = -1;
            sum += nums[i];
            if(sum == 0)    ans = i+1;
            else if(preSum.count(sum) == 0)  preSum[sum] = i;
            else    ans = max(ans,i-preSum[sum]);
        }
        return ans;
    }

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