leetcode-每日一题2021.6.3 连续数组 前缀和+哈希表

题目

https://leetcode-cn.com/problems/contiguous-array/

思路

把0当作-1，连续数组和为0就是满足条件的。从前往后，出现不同的前缀和就把它存到哈希表中，出现相同的前缀和就计算现在的最大长度与i-mp[cur]之间哪个大，更新最大长度

代码

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        unordered_map<int,int> mp={{0,-1}};
        int cur=0,maxlength=0;
        for(int i=0;i<nums.size();i++){
            nums[i]==0?--cur:++cur;
            if(mp.count(cur))
                maxlength=max(maxlength,i-mp[cur]);
            else
                mp[cur]=i;
        }
        return maxlength;
    }
};