LeetCode 每日一题 525.连续数组

---

问题描述

简要思路

和昨天的题一样，前缀和 + 哈希表。

代码

public:
    int findMaxLength(vector<int>& nums) {
    
    
        int n = nums.size();
        int res = 0;
        unordered_map<int, int> m;
        int sum = 0;
        m[sum] = -1;
        for(int i = 0; i < n; i++){
    
    
            sum += nums[i] ? 1 : -1;
            if(m.count(sum)){
    
    
                res = max(res, i - m[sum]);
            }
            else m[sum] = i;
        }

        return res;
    }
};