Leetcode-剑指 Offer II 011. 0 和 1 个数相同的子数组

链接

剑指 Offer II 011. 0 和 1 个数相同的子数组

题目

给定一个二进制数组 nums , 找到含有相同数量的 0 和 1 的最长连续子数组，并返回该子数组的长度。

示例

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

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

说明

• 1 <= nums.length <= 10e5
• nums[i] 不是 0 就是 1

思路

把0看作-1，这样就可以转换成前缀和问题，计算出前缀和数组后，两个相同的元素之间的所有元素的和便为0（也就是0和1的个数相同）

C++ Code

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        //将0换成-1 变成前缀和问题
        vector<int> Sum(nums.size()+1,0);
        int temp=0;
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]==1) temp+=1;
            else temp+=-1;
            Sum[i+1]=temp;
        }
        unordered_map<int,int> map; //记录某个值首次出现的位置
        int ans=0;
        for(int i=0;i<Sum.size();i++)
        {
            if(map.count(Sum[i])==0) map[Sum[i]]=i;
            else ans=max(i-map[Sum[i]],ans);
        }
        return ans;

    }
};